tensorflow报错self._traceback = tf_stack.extract_stack()解决-20200609

参考原文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)