conda 清华源安装pytorch 定向错误

2020-04-08  本文已影响0人  昵称己存在

国内安装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

上一篇 下一篇

猜你喜欢

热点阅读