Pycharm 首次导入 Tensorflow出现问题的解决方法

参考Pycharm中加入Tensorflow出现问题的解决方法

 

在看本文前需要先了解一下本文所解决的问题，表现为两个方面：

1、在终端中导入tensorflow导入时并不提示 tensorflow 模块不存在问题，而是提示cudnn找不到，问题如下所示：

dell@dell-PowerEdge-T630:~$ python
Python 2.7.12 (default, Dec  4 2017, 14:50:18) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:126] Couldn't open CUDA library libcudnn.so.5. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_dnn.cc:3517] Unable to load cuDNN DSO
I tensorflow/stream_executor/dso_loader.cc:126] Couldn't open CUDA library libcufft.so.8.0. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_fft.cc:344] Unable to load cuFFT DSO.
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.so.8.0 locally
>>> 

解决方法：

在环境配置文件中添加变量。一般在终端中导入成功的话这一步应该已经有了，如果终端导入失败，这一步必须做。

#先打开环境配置文件
gedit ~/.bashrc  

#在文件中添加
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64

#生效
source ~/.bashrc

 

 

成功解决。

 

2、在pycharm中导入时并不提示 tensorflow 模块不存在问题，而是提示cudnn找不到，问题如下所示：

Using TensorFlow backend.
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:126] Couldn't open CUDA library libcudnn.so.5. LD_LIBRARY_PATH: /home/dell/pycharm-community-2016.3.2/bin:
I tensorflow/stream_executor/cuda/cuda_dnn.cc:3517] Unable to load cuDNN DSO
I tensorflow/stream_executor/dso_loader.cc:126] Couldn't open CUDA library libcufft.so.8.0. LD_LIBRARY_PATH: /home/dell/pycharm-community-2016.3.2/bin:
I tensorflow/stream_executor/cuda/cuda_fft.cc:344] Unable to load cuFFT DSO.
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.so.8.0 locally

解决方法

（1）配置pycharm的python 编译器

Files  --->  Settings  ---->  Project Interpreter

配置该页面下的 Project Interpreter, 选择的python编译器下需要包含tensor flow包。

如下所示：

 

（2）配置pycharm的环境变量

Pycharm下选择Run->Edit Configurations...

 

也就是这是上面的Enviroment下的Enviroment Variables。添加LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64到还击你该变量。如下图：

 

 

至此就解决完全了。