could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

在进行深度学习训练时遇到'could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR'的问题,这通常是由于GPU资源不足导致。尽管可以切换到CPU继续训练,但速度大幅下降。为解决此问题,只需在导入库后添加代码,限制初始GPU内存分配,以降低显存占用,从而避免错误。
摘要由CSDN通过智能技术生成

当我们深度学习做训练的时候,偶尔会发生这种情况,我把源错误贴出来:

Epoch 1/16
2019-09-11 09:34:11.000335: E C:\users\nwani\_bazel_nwani\mmtm6wb6\execroot\org_tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:455] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2019-09-11 09:34:11.000570: F C:\users\nwani\_bazel_nwani\mmtm6wb6\execroot\org_tensorflow\tensorflow\core\kernels\conv_ops.cc:713] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms) 

Process finished with exit code -1073740791 (0xC0000409)

就是这么个事,按照我的理解,这是你使用GPU来训练网络的时候显卡不够用了,但是这个时候如果用CPU还是可以的,但是CPU和GPU的训练速度差的简直不是一星半点,我这里差了有十倍,解决方法只需要在import后边添加两行代码,让占用的显卡内存一开始不要那么高:
如下,只有最后两行有用

import cv2
‘’‘’‘’
import各种
‘’‘’‘’
import matplotlib.pyplot as plt

config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth&#
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值