Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

keras 运行yolo v3 出现Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED问题，
在代码最开始加入以下代码：

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)

亲测有用，原因可能是GPU占用问题导致的。
参考：
https://blog.csdn.net/weixin_42769131/article/details/88848478