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我选择的版本目前来说太高了:)