查看CUDA版本
1.查看环境当前安装的cuda版本
nvcc -V #这个是最准确的
nvcc --version #这两个都可以
2.查看当前服务器配置支持cuda最高版本,可向下兼容(表格右上角)
nvidia-smi
下载torch包
1.上官网,查看对应的torch版本(可略)
2.然后来这里:https://download.pytorch.org/whl/torch_stable.html 下载对应的torch
3.下载后拖到服务器中,运行:
pip install torch.名字 #可以在后面加上 -i 国内镜像源,在我的合集第一篇