1. 程式人生 > >Ubuntu 18.04下Pycharm匯入tensorflow時報Import Error的原因及解決方法

Ubuntu 18.04下Pycharm匯入tensorflow時報Import Error的原因及解決方法

最近剛轉到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.

出現的原因可能為:

  1. 沒有安裝Cuda或安裝了錯誤的版本。在一開始我安裝的是最新發布的Cuda10.0, 安裝後報錯。
  2. 直接使用桌面圖示而不是Shell啟動Pycharm。這樣會導致Pycharm無法匯入環境變數配置檔案~/.bashrc。因此出現了可以在Shell中的Python環境匯入但不能在Pycharm環境中匯入的情況。

針對以上兩種情況的解決方法為:

  1. 建議安裝Cuda9.0版本並在~/.bashrc中設定Cuda相關的環境變數。安裝教程在此不再贅述。
  2. 使用Shell命令: pycharm進行啟動
  3. 在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的環境變數。