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

本文详细介绍了在PyCharm中使用TensorFlow遇到cuDNN库未找到错误的解决方法,包括配置环境变量和PyCharm解释器设置,适用于Linux环境下PyCharm与TensorFlow的兼容性问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参考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到还击你该变量。如下图:

 

 

至此就解决完全了。

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值