国内安装pytorch官网上面的安装方式,容易在下载过程中中段。
解决方法是添加清华的镜像。 Following https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
这个时候安装的命令也要变换:
原来:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
改为:
conda install pytorch torchvision cudatoolkit=10.1
-c pyotrch
含义是制定为pytorch 的channel,会把清华源的channel覆盖。
这种说法是错误的。其实在修改.condarc
的时候,已经指定了pytorch
为清华的镜像。
这个时候安装的pytorch有时候会与制定的版本不一致,甚至是下载cpu版本的pytorch(可以看安装时显示的信息)
解决方法是直接指定channel.
conda install pytorch torchvision cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
也可能有下载问题。
能够在https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/ 看到一些pytorch 官网上面没有明说的版本。 比如pytorch10.1-cuda10.0
网友评论