解决Eclipse中pydev环境下tensorflow和keras不支持CUDA链接库

1.测试是否tensorflow和keras是否支持CUDA链接库  

import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) from keras import backend as K import keras print keras.__version__ print K.tensorflow_backend._get_available_gpus()

2.eclipse中如果 报：

Couldn't open CUDA library libcudnn.so

Couldn't open CUDA library libcufft.so

说明不支持

3.解决的方法

1.打开Windows->Preferences->PyDev->Interpreters->Python Interpreter

2.选择第二栏选项卡中的Environment，选择new

3.通过new，新建2个Variable，

CUDA_HOME=/usr/local/cuda

LD_LIBRARY_PATH=${CUDA_HOME}:/usr/local/cuda/lib64:usr/local/cuda/extras/CUPTI/lib64

*：variable的值根据个人电脑上的配置设定

这个选项其实就是设置eclipse的自定义环境变量

重新运行程序，加载成功。