【安装pytorch1.0 + cuda10.1】问题：ImportError:/usr/lib/libcudart.so.10.0:version 'libcudart.so.10.0' not..

最新推荐文章于 2024-01-13 10:22:26 发布

原创最新推荐文章于 2024-01-13 10:22:26 发布 · 7.1k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#安装软件 #pytorch #cuda

问题专栏收录该内容

10 篇文章

订阅专栏

本文讲述在Ubuntu下安装pytorch1.0 + cuda10.1 + python3.6，运行torchvision时出现问题。因官网无cuda-10.1选项，擅自修改安装命令导致一系列问题，随后详细介绍了不同问题对应的解决步骤，最终成功解决。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在ubuntu下安装pytorch1.0 + cuda10.1 + python3.6 ，运行torchvision时出现问题

libcudart.so.10.0: cannot open shared object file: No such file or directory

原因：我的Ubuntu下安装的是cuda-10.1，但安装pytorch时官网没有cuda-10.1的选项，只能安装cuda-10.0版的pytorch，另外我擅自把官网给的安装命令

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

中的10.0修改为10.1，因此在出现上述问题之后还有之后的一系列问题。。

解决步骤：

1. 如果你按照官网命令安装，出现上述问题后，输入下面三条命令即可，参考自这位博主

sudo ln -s /usr/local/cuda-10.1/lib64/libcudart.so.10.1 /usr/lib
sudo ln -s /usr/local/cuda-10.1/lib64/libcudart.so.10.1 /usr/lib/libcudart.so.10.0
sudo ldconfig

2. 如果你跟我一样修改为10.1了，在执行第一步之后会出现

ImportError:/usr/lib/libcudart.so.10.0:version 'libcudart.so.10.0' not..

参考自github，需要重新安装 cudatoolkit-10.0，执行以下命令

conda install -c anaconda cudatoolkit==10.0

3. 如果执行第二步时出现 PackageNotFoundError: Packages missing in current channels...

那是因为包的名字不对，则首先执行

anaconda search -t conda cudatoolkit

然后找到合适的安装包名称如 anaconda/cudatoolkit，接着执行

anaconda show anaconda/cudatoolkit

会显示多个版本，选择合适的版本，修改并执行显示的最后一行命令

conda install --channel https://conda.anaconda.org/anaconda cudatoolkit==10.0.130

4. 如果执行第三步跟我一样没有指定版本，不知道为啥pytorch对应的cuda版本就变成了9.0, 可能跟我一样在运行torchvision时还会出现Pytorch has CUDA Version=9.0 and torchvision has CUDA Version=10.0...

首先运行

pip show torch

找到pytorch的安装目录，然后在 {安装目录}/torch/version.py中，修改cuda变量为你刚刚安装的cudatoolkit版本，我的是

cuda = '10.0.130'

以上，输入 import torchvision 终于好了。。【鞠一把老泪】