本文主要是介绍【安装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 终于好了。。【鞠一把老泪】
这篇关于【安装pytorch1.0 + cuda10.1】问题:ImportError:/usr/lib/libcudart.so.10.0:version 'libcudart.so.10.0' not..的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!