could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED

训练、测试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等错误。

错误主要指向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)

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
cuDNN error: CUDNN_STATUS_NOT_INITIALIZED错误通常是由于多种原因引起的。根据引用\[1\]中的描述,这个错误可能与PyTorch的版本与cuda版本不匹配无关。引用\[2\]中提到,这个错误可能是由于输入的label值超出了值域所导致的。在这种情况下,网络要求的label应该在0到8之间,但是输入的label值域不正确,因此导致了错误。另外,引用\[3\]中指出,这个错误也可能是由于模型输出有问题,例如关系类别数量定义错误导致的。因此,要解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED错误,你可以尝试以下几个步骤: 1. 确保PyTorch和cuda的版本匹配。 2. 检查输入的label值是否在正确的值域内。 3. 检查模型的输出是否正确,例如关系类别数量是否定义正确。 通过逐步排查这些可能的原因,你应该能够解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED错误。 #### 引用[.reference_title] - *1* *3* [RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED](https://blog.csdn.net/liaoningxinmin/article/details/119139840)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED【解决方法】](https://blog.csdn.net/weixin_47675950/article/details/120437087)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值