在运行程序的时候报错:
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)
然后在重新运行程序,成功解决。