[大坑]Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

最新推荐文章于 2020-12-24 15:54:20 发布

转载最新推荐文章于 2020-12-24 15:54:20 发布 · 241 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/aoru45/p/10811428.html

文章标签：

#python #运维

博客围绕TensorFlow运行时出现的CUDNN内部错误展开。博主尝试多种方法，如修改GPU运行config、降级cuda和cudnn版本均未解决。最终发现用docker安装tensorflow - gpu可运行但不便，用anaconda在虚环境指定版本安装tensorflow - gpu是较好办法，不过只能暂时解决问题。

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

报错tensorflow/stream_executor/cuda/cuda_dnn.cc:338] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

错误代码

2019-05-05 09:34:42.298865: I tensorflow/stream_executor/platform/default/dso_loader.cc:42] Successfully opened dynamic library libcublas.so.10.0
2019-05-05 09:34:42.469839: I tensorflow/stream_executor/platform/default/dso_loader.cc:42] Successfully opened dynamic library libcudnn.so.7
2019-05-05 09:34:43.086143: E tensorflow/stream_executor/cuda/cuda_dnn.cc:338] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2019-05-05 09:34:43.100978: E tensorflow/stream_executor/cuda/cuda_dnn.cc:338] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

解决办法

首先这是一个大坑，在网上找了很多办法，基本都是改一些GPU运行时候的config，我发现跑有些程序的时候并不奏效；也有是要求降级cuda和cudnn版本，我尝试了从cuda10.1一路降到9.0，cudnn也一直换，报错依旧在，仍然不奏效。

为了解决这个问题，我尝试用docker安装tensorflow-gpu发现是可以成功运行程序的，但是不太方便，我发现anaconda在虚环境里面使用conda安装tensorflow-gpu会自动安装该环境下的cuda和cudnn，并且不会对环境外已经安装的cuda和cudnn有影响，但是默认安装的版本比较低，所以，最好的办法是指定安装的版本。只需要指定tensorflow的版本即可，cuda和cudnn的版本会对应。

在虚环境下安装