pytorch使用清华源快速安装,解决官方默认通道安装缓慢的问题

问题发现:使用pytorch官方安装的命令(如下图)安装pytorch很慢

(有人建议conda切换清华的源,然后用conda install pytorch torchvision cudatoolkit=*.*命令(去掉了官方的-c pytorch)安装,但我却安装上了cpu版本的pytorch和torchvision???我还不明白为啥,只能麻烦点了)

问题解决

依然输入这个命令,当出现如下信息后

 Proceed ([y]/n)?    选择   n

(注:由于不同的初始环境,上图中的following packages包含的包可能有很大区别,大家做好记录,将对应的包都下载安装好,注意都要用conda 进行安装)

接着,到pytorch的清华源找到对应名称的文件(如下图)

 然后wget下载对应的连接,如:

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.4.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/torchvision-0.5.0-py37_cu101.tar.bz2

下载完成后,本地conda install 安装,如:

conda install --use-local pytorch-1.4.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
conda install --use-local torchvision-0.5.0-py37_cu101.tar.bz2

 即可安装完成。

如后期 import torch时遇到报错:ImportError: libmkl_intel_lp64.so: cannot open shared object file: No such file or directory

则  conda install mkl解决。

 

发布了25 篇原创文章 · 获赞 28 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/maitianpt/article/details/104272815