ubuntu+anaconda+离线pytorch环境搭建

Posted by HieDean on March 19, 2020

官网下载anaconda

得到

因为用的是学校的设备,所以要scp filename hostname@hostIP上传至服务器 服务器端执行sh Anaconda3-2020.02-Linux-x86_64.sh即可

然后还要配置环境变量

echo 'export PATH="~/anaconda3/bin:$PATH"' >> ~/.bashrc

source ~/.bashrc

卸载anaconda

删除整个anaconda3文件即可 rm -rf anaconda3

下载pytorch之前先查看cuda和cudnn版本

cuda 版本

cat /usr/local/cuda/version.txt

nvcc -V

cudnn 版本(命令执行后得到的三个数字即cudnn版本)

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

(前提是保证机器上已经安装好了cuda和cudnn)

离线下载pytorch

参考https://www.cnblogs.com/darkknightzh/p/12000809.html

由于使用的是学校的机器,所以先本地下载安装包,再上传

  1. conda官网下载pytorch:https://anaconda.org/pytorch/pytorch/files(注意cuda和cudnn版本)

  2. conda官网下载torchvision:https://anaconda.org/pytorch/torchvision/files

  3. conda官网下载cudatoolkit:https://anaconda.org/anaconda/cudatoolkit/files

下载完成并上传至服务器后

conda install --offline pytorch-1.3.1-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2

conda install --offline torchvision-0.4.2-py37_cu101.tar.bz2

conda install --offline cudatoolkit-10.1.243-h6bb024c_0.tar.bz2

这三大件安装完成后,我按照参考文章中的说法,进行了一次 conda install pytorch torchvision cudatoolkit=10.1

结果可以看到图中torchvision(需要回退版本)和cudatoolkit仍然需要下载一遍

可是学校的机器流量不能这么挥霍啊!!!

于是我没有回退torchvision,也没有重新下载cudatoolkit,而是用conda install将其他包一个一个下载了下来(建议pillow使用6.1版本)

反正最后能使