背景
在PyCharm中運行tensorflow時,發現提示ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory這樣的錯誤。
問題原因分析
本地代碼在遠程服務器上解釋執行,但尋找外部依賴依然由本地框架完成,需要手工指定遠程服務器上安裝的第三方庫(python解釋器自己找不到的那種)。
解決方案
在PyCharm中,點擊右上角Edit Configuration或Add Configuration,點擊Python,界面如下圖:
在Environment Variables中添加兩個變量:
LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
CUDA_HOME=/usr/local/cuda-9.0/
注:以上變量的值,根據個人配置的cuda環境變量進行調整。
若想在PyCharm的Python Console中輸出結果,則找到下圖中的界面,在Environment Variables中添加和上面一樣的配置
點擊OK後,重新run,將不會出現錯誤提示,完美解決!