出錯的具體表現:
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