解决Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR问题

原因:显存不足(一般情况下,如果不加以限制,很多深度学习框架的代码在运行的时候会申请整个显存空间(即便它不需要这么多的资源,但是它申请了之后就不允许其他的程序使用),所以在这种状态下运行代码,就会出现显存不够用的问题(因为还有其他的程序或操作需要显存资源))
解决方法:在运行脚本中添加下述代码

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

猜你喜欢

转载自blog.csdn.net/qq_28057379/article/details/115088287