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

在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 终于好了。。【鞠一把老泪】