could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED

电玩女神 2022-03-22 10:38 242阅读 0赞

训练、测试Tensorflow、Keras代码时，出现could not create cudnn handle: CUDNN\_STATUS\_NOT\_INITIALIZED、error retrieving driver version: Unimplemented: kernel reported driver version not implemented on Windows、could not destroy cudnn handle: CUDNN\_STATUS\_BAD\_PARAM等错误。

![2019012809455912.png][]

错误主要指向cudnn，但是CUDA版本和cudnn版本是符合当前tensorflow要求的，因此只能是GPU占用问题导致的。解决方法如下：

tensorflow 框架下设置GPU按需分配：

import tensorflow as tf
    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True
    with tf.Session(config=config) as sess:
        ...

keras框架（Tensorflow backend） 设置GPU按需分配：

import tensorflow as tf
    from keras import backend as K
    config = tf.ConfigProto()
    config.gpu_options.allow_growth=True
    sess = tf.Session(config=config)
    K.set_session(sess)

\*\* Tensorflow 2.0 设置GPU按需分配方式（没有session）：

gpus = tf.config.experimental.list_physical_devices('GPU')
    for gpu in gpus:
        tf.config.experimental.set_memory_growth(gpu, True)

[2019012809455912.png]: /images/20220322/b00abdce7b8540028592a6c834e9e8cc.png

发表评论取消回复

表情：

评论列表（有 0 条评论，242人围观）

还没有评论，来说两句吧...

相关阅读

相关 Could not create the Java Virtual Machine

笔者第一次学习使用java。安装好了jdk和Eclipse，配置好PATH JAVA\_HOME CLASSPATH 后，在Eclipse中新建java project之后

小咪咪/ 2024年02月17日 17:24/ 0 赞/ 73 阅读

相关报错解决：could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

![20191009191333910.png][][日萌社][Link 1] [人工智能AI：Keras PyTorch MXNet TensorFlow Pad

迈不过友情╰/ 2023年06月16日 01:56/ 0 赞/ 60 阅读

相关 Tensorflow训练出现could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 错误

Tensorflow1.10 Gpu:2080 cuda:9 问题：执行训练的时候，就算batchsize设置为1，也会出现一下问题 could

忘是亡心i/ 2023年05月30日 04:54/ 0 赞/ 4 阅读

相关 Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

莫名其妙的报错： in tensorflow/stream\_executor/cuda/cuda\_dnn.cc(953): ‘cudnnSetDropoutDescrip

ゝ一纸荒年。/ 2022年10月14日 10:49/ 0 赞/ 163 阅读

相关【显存不足完美解决方案】Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

一、问题出现在使用TensorFlow或者Keras准备进行网络训练时，有时候会出现如下报错： Could not create cudnn handle:

布满荆棘的人生/ 2022年10月06日 09:50/ 0 赞/ 214 阅读

相关 Linux FTP 553 Could not create file.

问题： FTP put 时候报错解决办法： /abaptest/soft/ 为你要写/上传文件的目录

朱雀/ 2022年08月05日 13:29/ 0 赞/ 208 阅读

相关 Could not create JarEntryRevision

前些天刚好开始Struts的学习，但是在配置Struts 2的时候就遇到了很大的障碍，我使用的Struts 2版本是2.2.1（即Struts 2.2.1），我参考了《精

你的名字/ 2022年06月15日 12:42/ 0 赞/ 212 阅读

相关 could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED

训练、测试Tensorflow、Keras代码时，出现could not create cudnn handle: CUDNN\_STATUS\_NOT\_INITIALIZE

电玩女神/ 2022年03月22日 10:38/ 0 赞/ 243 阅读

相关 Cannot create PoolableConnectionFactory (Could not create connection to database server.

1.版本改到最高 <dependency> <groupId>mysql</groupId> <artifactId>mysql-connector-java</ar

傷城~/ 2022年01月09日 06:41/ 0 赞/ 325 阅读

相关在使用keras时报错显示tensorflow 1.4 could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR的解决办法

在代码中增加下面一些代码即可 import os import tensorflow as tf from keras.backend.tensorf

悠悠/ 2021年11月05日 14:26/ 0 赞/ 283 阅读