在anconda上安装pytorch的gpu版教程(一天时间遇到的坑),本来以为一会就安完了,没想到兜兜转转一天时间,记下这耻辱的一刻,顺便分享一下。
首先,我的需求是在anconda上安装pytorch的gpu版本。
遇到的问题构成了死循环。

1.在线安装

一句命令搞定,丝滑连招走起 。
我的anconda使用了清华源但还是安不上,但还是总结一下吧,欧皇们可以试试。
1.清华镜像源方案
该方案就是将官网的下载路径,改到清华镜像源的路径,先用activate激活自己之前创建的环境后,再命令行输入以下代码:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

2.然后再去pytorch官网复制下载代码,记得删掉后面的-c pytorch,因为它表示从官网进行下载,不从清华镜像源下。
官网查看版本的网址是:https://pytorch.org/

conda install pytorch torchvision cudatoolkit=10.2

2.离线安装

清华镜像源本地下载
(linux版本)https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/?C=M&O=A或https://anaconda.org/pytorch/pytorch/files?version=1.3.0
(windows版)https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=A
离线安装代码:

conda install --offline pytorch-1.9.0-py3.8_cuda10.2_cudnn7_0.tar.bz2

安装代码需根据自己下载的版本来修改!offline 后面那一串都是下载的文件名,我这里示意的是linux版本的,win10版的得看你自己下载好的文件名是什么

离线安装时,记得用cd命令把当前目录改到下载的离线文件所在目录

torchvision也要单独安装:

conda install --offline torchvision-0.10.0-py38_cu102.tar.bz2

离线安装后,还需要重新利用conda在线安装一次,要用它来对相关的包来进行在线匹配(也就是根据安装好的pytorch来匹配那些对应的好多库,不是说再下载一遍pytorch)。
但是一定要注意了!!!在利用官网代码时conda install pytorch torchvision cudatoolkit=9.2 -c pytorch一定要删掉后面的-c pytorch,因为它代表直接从官网去下载,不会与本地安装好的匹配,还会全部重新下载。
所以改成:

conda install pytorch torchvision cudatoolkit=10.2

补充:
如果cudatoolkit,mkl之类的也下载很慢,也可以用上面的方法进行离线安装。首先观察conda自动匹配推荐的文件名,然后去清华镜像源网站下载,然后离线安装,和上面一样。注意网站的文件夹:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/