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

一、环境

• Windows 10
• Tensorflow-gpu 2.0 正式版
• GTX 1660Ti

二、问题

使用卷积神经网络进行卷积运算的时候，发现运行失败，报错如下：

tensorflow.python.framework.errors_impl.UnknownError: 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. [Op:Conv2D] 

大致一看，问题应该出在 cuDNN 上。

三、解决方案

首先考虑的是 cuda 和 cuDNN 的版本问题。
尝试了几次，问题玄学般的得到了解决。。。
最后安装的版本是：

1. cudnn-9.0-windows10-x64-v7.6.0.64
2. cuda_10.0.130_411.31_win10

在安装这些软件的时候一定要根据自己的显卡，选取正确的版本进行安装。