问题描述
在用pip安装Tensorflow时,或者安装完import tensorflow as tf
调用时,提示libcublas.so.9.0: cannot open shared object file: No such file or directory
的错误。
出现原因
-
Cuda-9.0
未安装或安装不正确 - 未添加环境变量或添加完未更新
- Tensorflow版本不支持当前Cuda或Cudnn版本
解决办法
终端输入
nvcc -V
查看系统是否已安装Cuda-9.0
,未安装则根据教程安装。检查环境变量是否添加:
终端输入:
sudo gedit .bashrc
在末尾加入:
export PATH="$PATH:/usr/local/cuda-9.0/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64"
一定要更新环境变量!
source .bashrc
按上述操作后,如果仍未解决问题,则继续下一步。
- 检查
/usr/local/cuda-9.0/lib64
路径下是否有libcublas.so.9.0文件。
如果有,在终端输入:
sudo ldconfig /usr/local/cuda-9.0/lib64
上述三部之后仍然提示出错,则需要考虑安装的Tensorflow版本是否支持当前安装的Cuda和Cudnn版本,如果不支持,则需要回滚至旧的版本。
sudo pip --upgrade tensorflow-gpu==型号