导读

在使用conda安装pytorch的时候下载特别慢,有时候下载到一半了还会断掉,然后还需要重新下载才行,很难受。

比如我们要安装pytorch1.3.1的版本,安装命令如下

conda install pytorch=1.3.1 torchvision=0.4.2 -c pytorch
解决办法
  • 设置清华的conda
# ubuntu
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
#look setting info
conda config --show-sources
  • 先下载好pytorch的安装包再安装

清华大学pytorch镜像
去清华大学的镜像网站下载对应系统版本的pytorch,需要注意你的系统环境和cuda以及cudnn的版本,下载的时候可以用IDM进行多线程下载2分钟就能下好。

下载好之后再进行安装,安装命令如下

conda install [your .bz2 directory]

安装torchvision

conda install torchvision
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐