tensorflow-gpu出错:ImportError: libcublas.so.9.0: cannot open shared object file

出错的具体表现:

tensorflow-gpu在终端可以正常运行,但是在pycharm中运行报错。(LInux)

具体的错误信息:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory.
Failed to load the native TensorFlow runtime.

解决方法:

  1. 查询 tensorflow-gpu的版本是否和 cuda, cuddn 版本匹配。(我的配置是 tensorflow-gpu 1.8.0 + cuda 9.0 + cuddn 7.0)。若不匹配,解决匹配问题。
  2. 检查环境变量是否添加(这是我这边报错的主要原因

在根目录下执行:

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"

注:/usr/local/cuda<对应下载的版本,自行修改文件夹名>/

  1. 更新环境变量
source .bashrc
  1. 检查 “/usr/local/cuda-9.0/lib64” 文件夹下是否有 libcublas.so.9.0 文件,若有,在终端执行:
sudo ldconfig /usr/local/cuda-9.0/lib64

以上,成功解决上述问题。
总结:

  • tensorflow-gpu, cuda, cuddn三者的版本匹配问题
  • cuda 对应的系统环境变量设置

感谢分享:https://www.jianshu.com/p/6c649b1969c7?utm_source=oschina-app

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章