TF2问题解决——Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

1.所面对的问题

当我们在使用tensorflow进行机器学习模型搭建和训练的时候，经常会遇到以下问题：

---

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

---

2.问题出现的原因

在TF代码运行的时候，程序会申请很多显存，可是我们显卡的显存不能满足，因此出现了这种错误。

3.问题的解决办法

此时我们需要在代码的开始添加以下代码：

os.environ['TF_XLA_FLAGS'] = '--tf_xla_enable_xla_devices'
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

此时，程序便会按需申请显存空间，不会出现显存不足的情况。