问题发现:使用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解决。