运行tensorflow -gpu出现 CUDNN_STATUS_ALLOC_FAILED

背景介绍

我是cuda 10.2版本，用的conda install 安装的tf，使用yolo v4的时候出现这个问题。

解决办法

在代码最开头加入如下代码。

import tensorflow as tf

config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)

亲测可用。
参考文章：
https://blog.csdn.net/weixin_44754046/article/details/97663626