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

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