问题背景
最近在安装pytorch的时候总遇到网络问题,在线安装动不动就失败,即使更换了清华源也会出现一样的问题,于是打算先下载pytorch的安装文件,再进行离线安装,简单记录一下。
参考博客:https://www.cnblogs.com/darkknightzh/p/12000809.html
下载pytorch和torchvision
去清华的镜像网站下载pytorch和torchvision,根据你需要的环境下载不同的版本,我下载的是pytorch-1.6.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2和torchvision-0.7.0-py37_cu101.tar.bz2
清华大学镜像
下载cudatoolkit
去conda官网下载cudatoolkit,我下载的是cudatoolkit-10.1.243-h6bb024c_0.tar.bz2
cudatoolkit下载地址
安装pytorch、torchvision、cudatoolkit
conda install --offline pytorch-1.6.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
conda install --offline torchvision-0.7.0-py37_cu101.tar.bz2
conda install --offline cudatoolkit-10.1.243-h6bb024c_0.tar.bz2
安装完毕以后运行安装命令来匹配之前离线安装好的东西
conda install pytorch torchvision cudatoolkit=10.1
检验是否安装成功
出现以上结果即为安装成功,torch.cuda.is_available() 代表可以使用GPU,如果是false,意味着只能使用CPU。