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

我先从镜像安装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.csdn.net/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就不会报错了。

猜你喜欢

转载自blog.csdn.net/Jeff_zjf/article/details/87610460