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

白话先生NIT

于 2020-07-02 10:57:48 发布

阅读量322

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/liuweizj12/article/details/107079901

本文介绍如何使用TensorFlow和Keras检测CUDA链接库的支持情况，并提供了解决Eclipse中CUDA库加载失败问题的详细步骤，包括设置自定义环境变量以确保程序正确加载CUDA库。

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

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的自定义环境变量

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