conda install命令无法安装pytorch

由于网络问题,直接采用conda install命令可能无法直接下载对应的cuda包。

方法一:采用pip命令替代

步骤1:切换pip的源为国内源:

若只是临时切换:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

若永久切换pip源:

pip install pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

若pip 默认源的网络连接较差,可以临时使用镜像站来升级 pip:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pip -U

临时切换和永久切换二选一即可。

步骤二:去pytorch官网下载

去官网选择对应的命令和cuda版本下载。
pytorch官网

方法二:conda的pytorch官网

torch官网
可以选择对应的版本:
选择版本
单击即可下载。

本地下载安装这个包即可:

conda activate GSNet
conda install --use-local ~/Downloads/linux-64_pytorch-1.8.0-py3.7_cuda11.1_cudnn8.0.5_0.tar.bz2

conda安装

这种方法不止针对pytorch其他方法也可以下载其他包也是在https://anaconda.org网址中查找。

可能出现的问题:

问题一:在import torch后:
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/lu/anaconda3/envs/GSNet/lib/python3.7/site-packages/torch/__init__.py", line 195, in <module>
    _load_global_deps()
  File "/home/lu/anaconda3/envs/GSNet/lib/python3.7/site-packages/torch/__init__.py", line 148, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  File "/home/lu/anaconda3/envs/GSNet/lib/python3.7/ctypes/__init__.py", line 364, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: libmkl_intel_lp64.so: cannot open shared object file: No such file or directory

首先找到这个文件所在的位置:

locate libmkl_intel_lp64.so

找到文件所在的位置
然后放到.bashrc文件中去加入这一行:

export LD_LIBRARY_PATH=/home/lu/anaconda3/lib:$LD_LIBRARY_PATH

可能的原因是它会去LD_LIBRARY_PATH这个路径下找对应的so文件,如果找不到的话就没有。所以需要显式的指定这个路径。

问题二:

在命令行中成功运行后,在pycharm中运行可能也会存在问题。
pycharm中执行的问题
pycharm似乎不会在LD_LIBRARY_PATH路径下查找对应的so文件,而是将其/usr/lib/,/usr/lib64/,/usr/local/lib64/。
解决方法是直接复制到/usr/lib/,/usr/lib64/,/usr/local/lib64/下。

cp /home/lu/anaconda3/lib/libmkl_intel_lp64.so /usr/lib/
cp /home/lu/anaconda3/lib/libmkl_intel_lp64.so /usr/lib64/
cp /home/lu/anaconda3/lib/libmkl_intel_lp64.so /usr/local/lib64/

然后即可成功运行。

  • 8
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值