Pycharm中加入Tensorflow出現問題的解決方法

初次使用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設置完成,再運行就不會出錯了。

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