为了在pycharm导入tensorflow模块,需要做如下设置:
file-settings-project interpreter-project interpreter-add local
Name:tensorflow
Interpreter path:选择tensorflow/bin/python路径
在命令行运行无错误,在pycharm下报如下错误:
Couldn't open CUDA library libcudnn.so. LD_LIBRARY_PATH:
I tensorflow/stream_executor/cuda/cuda_dnn.cc:3448] Unable to load cuDNN DSO
等错误
打开pycharm文件夹bin下pycharm.sh, 添加:
LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"