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

前言

需要使用tensorflow-gpu进行测试GPU性能，之后在RTX2080Ti上安装成功python3.6.7以及CUDA10.0以及cudnn7.3以后 安装成功tensorflow-gpu1.12.0

pip3 install tensorflow-gpu

ok！安装成功后，发现import失败

解决之道

网上搜了一圈，找到几种办法：

1、修改链接路径

因为之前安装过cuda9 不知是否这个原因导致它竟然去找了cuda9的库

[root@localhost ~]# locate libcublas.so.9.0
/home/cuda_9/lib64/libcublas.so.9.0
/home/cuda_9/lib64/libcublas.so.9.0.176

因此，我当前的解决办法是直接修改LD_LIBRARY_PATH环境变量 让你成功找到
测试一下

vim ~/.bashrc
#写入下面内容
export CUDA_HOME=/usr/local/cuda-10.0
export PATH=${PATH}:${CUDA_HOME}/bin
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${CUDA_HOME}/lib64:/home/cuda_9/lib64/
#结束
source ~/.bashrc

测试成功！

[root@localhost ~]# python3
Python 3.6.7 (default, Nov 29 2018, 19:16:28) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-28)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> 

2、修改对应程序版本

2.1 cuda10换成cuda9版本就行 对应的cudnn库也得改

个人觉得比上面的方法耗时 不取

2.2 换tensorflow的版本

#回滚到某个版本
pip install --upgrade tensorflow-gpu==1.4

我试了下，发现对于我的问题并没有解决，maybe我选择的版本目前来说太高了：）

Reference

[1] tensorflow图像分类
[2] ImportError: libcublas.so.9.0: