出错的具体表现:
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.
解决方法:
- 查询 tensorflow-gpu的版本是否和 cuda, cuddn 版本匹配。(我的配置是 tensorflow-gpu 1.8.0 + cuda 9.0 + cuddn 7.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"
注:/usr/local/cuda<对应下载的版本,自行修改文件夹名>/
- 更新环境变量
source .bashrc
- 检查 “/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