Linux服务器上添加了清华镜像也无法安装pytorch解决方案
创建虚拟环境到手动安装包
今天遇到加镜像也还是安装不了pytorch的问题,首次尝试手动下载并安装,仅此记录供大家参考。
建立虚拟环境
1.conda create -n [虚拟环境名称] python=3.8
2.conda activate [虚拟环境名称]
安装pytorch
1.(已经默认配置了清华镜像)
conda install pytorch torchvision torchaudio cudatoolkit=11.0
2.根据两个大文件的链接下载两个压缩文件到本地,然后传到服务器上同步代码的那个文件夹里。
在ssh session里输入
conda install pytorch-1.7.1-py3.8_cuda11.0.221_cudnn8.0.5_0.tar.bz2
以及
conda install cudatoolkit-11.0.221-h6bb024c_0.conda
以及最后
conda install pytorch torchvision torchaudio cudatoolkit=11.0
即可完成安装
3.可以看到在虚拟环境的site-packages文件夹里有pytorch包,文件夹路径 /home/xxx/anaconda3/envs/虚拟环境name/lib/python3.8/site-packages
但是貌似看不到cuda的路径,参考这篇文章
https://blog.csdn.net/Mr__George/article/details/102972958
4.检查是否安装成功