kerasGPU训练模型问题:Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

当在训练模型时遇到'Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR'的问题,本文提供了一种针对keras用户的解决方案。不需要在创建会话时添加代码,而是通过修改tensorflow的session.py源码来解决。具体步骤包括删除~/.nv/文件夹,以及在Session类的初始化函数中插入特定代码。完成修改后,错误将被消除。
摘要由CSDN通过智能技术生成

无法创建cudnn句柄解决方法

Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
训练模型遇到无法创建cudnn句柄问题,经过多方查阅资料,发现以下几种解决方法

sudo rm -rf ~/.nv/

本人执行过,木有成功

在创建会话是添加两行代码

config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值