安装torch1.1.0过程

方式一

从清华镜像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

在这里插入图片描述

torch和torchvision对应版本

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值