解决Ubuntu+Eclipse+pydev+tensorflow CUDA链接库不能加载

前端之家收集整理的这篇文章主要介绍了解决Ubuntu+Eclipse+pydev+tensorflow CUDA链接库不能加载前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

现象:使用eclipse run python程序时,Console中错误提示

Couldn't open CUDA library libcudnn.so

Couldn't open CUDA library libcufft.so

经过排查,cuda,cudnn均已正确安装,环境变量也正常配置,同时,在系统终端中运行该py文件无报错,于是怀疑是eclipse设置有问题,随即通过如下方式解决

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

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


How to install cuDnn

猜你在找的Ubuntu相关文章