问题描述:
我是之前下载了一份预训练的代码,然后之前跑通了,没问题,但是时隔一个月以后我再次用这个代码就出现了这样的错误
Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
经过查询发现了一个解决方案,但是并不适用我的代码
人家的解决方案是这样的:
在要执行的python文件添加如下代码即可。
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
————————————————
原文链接:https://blog.csdn.net/pkuyjxu/article/details/89402298
在我的代码里边tensorflow.compat.v1这个部分是报错的,因此我改成了这样:
from tensorflow import ConfigProto
from tensorflow import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
这样的话就没有这个错误了,成功的得到了结果。
出现这个问题的具体原因我没有仔细了解,如果您知道,方便的话,可以告知我一下,谢谢!