载入tensorflow出现错误,ImportError: libcublas.so.9.0

错误:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
问题:找不到cuda9.0的版本。
出现该错误的主要原因:cuda未安装或者cuda的版本有问题

而在此之前,我刚刚安装了tensorflow1.5.1版本,所以原有8.0版本的cuda不能用了。
官网:https://developer.nvidia.com/cudnn
在官网下载对应版本,我这里选择的是cuDNN v7.0。
下载后,
解压:tar zxvf cudnn-8.0-linux-x64-v6.0.tgz
安装:cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp -a lib64/libcudnn* /usr/local/cuda/lib64

其余版本对应:
Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0→cuDNN v5.1 Library for Linux
Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0→cuDNN v6.0 Library for Linux
[Download cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0]→cuDNN v7.0.5 Library for Linux

该方法只能在cuda版本正确情况下,安装正确版本的cudnn,没有解决问题。
由于本服务器的系统里有很多老环境的代码在运行,无法及时更新cuda,因为我决定还是回退到tensorflow1.4版本

你可能感兴趣的:(linux,python,tensorflow)