“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法

最近在使用TF2.0。运行程序出现以下错误。

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

 

一开始怀疑是CUDA和CuDNN配置错误（要求版本匹配）。反复试验后，还是有这个错误。
最后发现可能是GPU内存不足造成的。需要在程序前加以下一段代码：

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

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

 

意思是对GPU进行按需分配。
主要原因是我的图像比较大，消耗GPU资源较多。但我的显卡（RTX2060）显存只有6GB，所以会出现这个错误。这个错误提示有很大的误导性，让人一直纠结CUDA和CuDNN的版本问题。故在此立贴，以免后人重蹈覆辙。

参考：

    https://github.com/tensorflow/tensorflow/issues/24828
————————————————
版权声明：本文为CSDN博主「史丹利复合田」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/tsyccnh/article/details/102938368