tensorflow-gpu error：CUDNN_STATUS_ALLOC_FAILED或者self._traceback = tf_stack.extract_stack()

tensorflow-gpu error：CUDNN_STATUS_ALLOC_FAILED或者self._traceback = tf_stack.extract_stack()

 

在有些情况下，因为深度学习框架版本更新，细节的变动会使我们的代码最初对应修改：

报错信息（出现其中一种）：

1.Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
2.self._traceback = tf_stack.extract_stack()

若文本导入tensorflow:则在红色部分后添加以下三行信息：

import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

若文本导入keras:则在红色部分后添加以下一行信息：
from keras import backend as K
K.set_session(sess)

若二者都已导入，可直接如下：

import tensorflow as tf
from keras import backend as K
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
K.set_session(sess)
————————————————
版权声明：本文为CSDN博主「Draw_1999」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/Draw_1999/article/details/95051490