方式一
从清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/里面,下载需要的pytorch版本,然后用命令安装:
conda install pytorch-1.1.0-py3.7_cuda10.0.130_cudnn7.5.1_0.tar.bz2
遇到错误:
ImportError: libmkl_intel_lp64.so: cannot open shared object file: No such file or directory
原因是libmkl_intel_lp64.so是一个动态链接库文件,系统找不到。
解决办法:
在用户目录下找到libmkl_intel_lp64.so,并将此路径添加到环境变量中,打开.bashrc文件,在最后一行添加
export LD_LIBRARY_PATH=/home/zjx/anaconda3/lib:$LD_LIBRARY_PATH
更新环境变量配置文件并显示
source ./.bashrc
echo $LD_LIBRARY_PATH
如果能正确显示刚才添加的路径,则设置成功。
但是import torch又会产生另一个错误
原因应该是安装的torch和cuda版本不匹配
因此采用另一种方式进行安装
方式二
conda install pytorch==1.1.0
conda install pytorch==1.1.0