安装pytorch与服务器cuda版本不匹配问题解决
- 查询Linux服务器安装的cuda版本 ,注意区分服务器当前安装的cuda版本和最高支持的cuda版本;
nvcc -V //查询的是服务器当前安装的cuda版本
nvidia-smi //查询的是服务器最高支持的cuda版本
- 需要同时匹配pytorch、cuda和python版本 ,三者的对应关系表格我参考的博文链接如下:https://blog.csdn.net/m0_59118857/article/details/132379012
- 源码安装pytorch的whl文件 ,官网下载链接如下:https://download.pytorch.org/whl/torch_stable.html
- 找torch与torchvision、torchaudio的对应关系,然后分别安装三者 ,三者的对应关系表格我参考的博文链接如下(有规律,按照规律下载源码安装):https://blog.csdn.net/shiwanghualuo/article/details/122860521
- 在服务器命令行三次wget分别安装torch、torchvision、torchaudio;
//以pytorch安装为例
wget https://download.pytorch.org/whl/cu113/torch-1.10.2%2Bcu113-cp36-cp36m-linux_x86_64.whl --no-check-certificate
- pip install安装三份文件;
- 检验torch是否成功安装,验证过程如下,如果不报错就说明安装成功啦;