ubuntu系统上安装pytorch1.1.0

首先是安装MiniConda,
这个可以直接去清华镜像源上下载。
地址如下:MiniConda
下载完成之后,在输入类似于下面的指令,具体指令根据版本变化。

bash Miniconda3-4.5.4-Linux-x86_64.sh

在弹出的框里面,全部通过,直到遇到

Miniconda3 will now be installed into this location:
/home/XXX/miniconda3

  - Press ENTER to confirm the location
  - Press CTRL-C to abort the installation
  - Or specify a different location below

[/home/XXX/miniconda3] >>> 

在这里,需要输入你希望的文件路径:
如:env/miniconda3。
安装完成后,输入指令

 . XXX/Env/miniconda3/bin/activate

激活这个Conda环境。
之后就可以通过你的需要安装自己想要的包了。

在以下路径去下载pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2
国内镜像源
之后通过

conda install --use-local pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2

然后安装cuda10.0.130

conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

安装完成后,如:进入python环境,导入torch,出现以下报错。

>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/stu/laibb/Env/miniconda3/lib/python3.6/site-packages/torch/__init__.py", line 79, in <module>
    from torch._C import *
ImportError: libmkl_intel_lp64.so: cannot open shared object file: No such file or directory

可通过下面的方法解决:
非root用戶import torch遇到ImportError: libmkl_intel_lp64.so: cannot open shared object file: No such file
不过与之不同的是,我选择复制该文件到我的目录下。

cp libmkl-avx512.so ../XXX/Env/miniconda3/lib/

之后再导入torch,就可以运行了。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值