Tensorflow安装后报错:cannot open shared object file libcublas.so.9.0的解决办法

本文解决了一个常见的问题:使用GPU版本的TensorFlow时遇到的与CUDA版本不匹配导致的错误。作者详细记录了从镜像安装TensorFlow-GPU到因CUDA版本不符而出现的错误,再到如何正确安装CUDA以适配TensorFlow的过程。

我先从镜像安装gpu版本的tensorflow:

sudo pip install -i https://pypi.douban.com/simple  tensorflow-gpu

之后,在python下import tensorflow出错:

  • ImportError: libcublas.so.9.0: cannot open shared object file: No
    such file or directory

这是因为tensorflow与cuda之间有版本对应关系,我安装的是cuda-10,最新(截至2019.02)的tensorflow-1.11.0只支持cuda-9,所以会找不到libcublas.so.9.0。
版本对应关系:https://blog.youkuaiyun.com/omodao1/article/details/83241074

所以重装cuda-9.0:(可以不卸载cuda-10.0,但要改环境变量)
官网下载cuda-9.0的deb包

sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda-9.0
sudo gedit ~/.bashrc

把~/.bashrc中的
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/targets/x86_64-linux/lib
中的cuda-10.0改成cuda-9.0
source ~/.bashrc即可

此时再import tensorflow就不会报错了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值