深度大佬进修 发表于 2018-9-18 14:35:44

Error:ImportError: libcudnn when running a TensorFlow program

Error:ImportError: libcudnn when running a TensorFlow program

这个错误可能是由于没有安装cudnn只需安装合适的cudnn并添加环境变量即可。
通过下面网址下载：
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v5.1/prod_20161129/8.0/cudnn-8.0-linux-x64-v5.1-tgz
然后将文件解压分别放在下面文件夹
/usr/local/cuda/lib64
/usr/local/cuda/include
添加下面的环境变量（结合自己情况）
LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
CUDA_HOME=/usr/local/cuda
如果已经安装cudnn并且添加了环境变量还依然报错，那么可能是你的cudnn版本不适合，需要更换版本。

小虾米爱IT 发表于 2018-10-10 08:05:36

搜嘎，阿里嘎多

他改变了中国 发表于 2018-10-10 08:18:08

cudnn引发的惨剧

天使与魔鬼 发表于 2018-10-10 08:24:44

我是环境变量没添加

强人锁男 发表于 2018-10-10 08:25:45

解决了我的问题,3q

I_Like_AI 发表于 2018-10-11 08:08:04

原来是缺cudnn的库

黑崎一护 发表于 2018-10-11 08:12:08

环境变量小觑不得
页: [1]
查看完整版本: Error:ImportError: libcudnn when running a TensorFlow program