Tensorlfow2.Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

TF2.X遇到 Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
已经验证过的训练脚本，莫名出现这个问题。
一直在cudnn上纠结了很久

后来发现只需要在训练脚本前加上即可

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

来自
https://github.com/tensorflow/tensorflow/issues/24828