在使用conda对pytorch进行安装时,由于原镜像下载过慢,甚至终止的情况,可采用以下方案进行解决
1.添加清华园镜像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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(这个命令的意思是从channel中安装包时显示channel的url,这样就可以知道包的安装来源了)
2.开始安装pytorch,默认已经在pytorch官网拿到符合自己安装要求的命令了,例如:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
注意:要想使用添加的镜像源进行安装,则需要将命令最后的-c pytorch去掉;(因为这表示从原镜像中去下载)
3.所以,直接执行:
conda install pytorch torchvision torchaudio cudatoolkit=10.2
4.等待安装完毕即可!
|