初次使用Pycharm,加載Tensorflow,出現Cudnn找不到等問題。現在給出解決方案。
如果你配置正常,執行過程中就會出現一下界面(這是使用tensorflow ,cnn訓練的mnist數字識別的baseline)
一、設置python編譯器
File->Setting...->Project:**(我的項目名叫TF,所以是Project:TF)
也就是修改上面的Projiect Interpreter爲你tensorflow下Python.
我的tensorflow在安裝的時候設置了tensorflow安裝環境。所以使用下面的命令查找Python編譯器的路徑。
上面python的路徑就是:
/home/gaoxiaoya/anaconda2/envs/tensorflow/bin/python
複製這個到Project Intepreter。也就是點擊右面設置Add...添加路徑。
設置完上面的,只是導入tensorflow不會出錯。運行時還是會因爲沒有找到cudnn相關的庫出錯。
二:添加環境變量Environment variables
先查看下自己的系統環境變量 。命令:cat ~/.bashrc
裏面的export 就是一般裝完tensorflow會出現的系統變量。
Pycharm下選擇Run->Edit Configurations...
也就是這是上面的Enviroment下的Enviroment Variables。添加LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64到還擊你該變量。如下圖:
OK設置完成,再運行就不會出錯了。