Ubuntu 18.04下Pycharm匯入tensorflow時報Import Error的原因及解決方法
阿新 • • 發佈:2018-12-14
最近剛轉到Ubuntu上學習深度學習。在安裝開發環境時踩了許多坑。其中一個就是在使用Pycharm匯入tensorflow-gpu版時會報如下錯誤:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.
出現的原因可能為:
- 沒有安裝Cuda或安裝了錯誤的版本。在一開始我安裝的是最新發布的Cuda10.0, 安裝後報錯。
- 直接使用桌面圖示而不是Shell啟動Pycharm。這樣會導致Pycharm無法匯入環境變數配置檔案~/.bashrc。因此出現了可以在Shell中的Python環境匯入但不能在Pycharm環境中匯入的情況。
針對以上兩種情況的解決方法為:
- 建議安裝Cuda9.0版本並在~/.bashrc中設定Cuda相關的環境變數。安裝教程在此不再贅述。
- 使用Shell命令: pycharm進行啟動
- 在Pycharm的Run->Edit Configurations介面的Environment variables列表中加入Cuda的環境變數項,具體為:
LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64
(Cuda安裝路徑)。在File->Setting->Build, Execution, Deployment->Console->Python Console中加入Cuda的環境變數。