-
查看python、pytorch、cuda以及显卡对应的版本
-
nvidia-smi查看显卡驱动,若报错则需要安装显卡驱动

- 这里的CUDA-Version意思是等下要安装的CUDA的版本要<=13.1
-
nvcc -V查看是否安装了cuda,若报错,还需要去安装cuda工具包,这一步要注意版本匹配

本地下载之后进行安装,一路默认即可。完成之后再次用
nvcc -V命令检查:

-
登录pytorch官网,进行选择对应的版本进行安装

-
检查是否安装成功。python ->>
import torch->>torch.is_available()

ptyorch安装
Forrit2026-02-09 23:20
相关推荐
Είναι η κοπέλα18 小时前
模型量化完全指南:GGUF、GPTQ、AWQ 怎么选Soonyang Zhang2 天前
根据时戳从pytorch trace中获取kernel信息承渊政道2 天前
【从零开始大模型开发与微调:基于PyTorch与ChatGLM】(开源大模型ChatGLM使用详解)129Lab3 天前
电池热管理仿真的AI加速:用Python+PINN物理信息神经网络替代传统CFD的可行性探索Είναι η κοπέλα3 天前
llama.cpp 与 GGUF 格式:本地大模型的“裸引擎“头发够用的程序员3 天前
TensorRT 自定义算子插件实战(三):手搓 2×2 最大池化 customMaxpoolfor_ever_love__4 天前
PyTorch 张量与 autograd——自动求导怎么工作Είναι η κοπέλα4 天前
显存计算与模型选择:你的显卡能跑多大的模型海天一色y6 天前
模型部署的「通用语」:ONNX 转换Thomas.Sir6 天前
第26课:工业零部件外观缺陷检测系统:从学术Demo到产线工程的重构实战