tensorflow cuDNN 兼容性答案

【问题标题】：tensorflow cuDNN compatibilitytensorflow cuDNN 兼容性
【发布时间】：2019-03-10 10:41:56
【问题描述】：

我正在使用 keras 制作模型。

编译时，我的模型不起作用，并弹出错误消息：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.

我的电脑规格如下：

GPU：RTX2070，张量流版本：1.13.1， Python版本：3.6.5， CUDA：10.0， cuDNN：7.4.2

我尝试了 cuDNN 7.5.0 和此链接：cannot train Keras convolution network on GPU，但更改 cuDNN 版本对我不起作用。

所以，我尝试了这些代码：

>>>import tensorflow as tf
>>>a = tf.constant([1])
>>>b = tf.constnat([2])
>>>sess = tf.Session()
>>>with tf.device('/gpu:0'):
...    print(sess.run(a+b))
...
[3]

有效！有谁知道我为什么会遇到这个问题？

【问题讨论】：

标签： python tensorflow keras

【解决方案1】：

这个问题可能对https://github.com/tensorflow/tensorflow/issues/24828有帮助

尝试检查你有哪些版本的 cudnn 和 tensorflow。

【讨论】：

【解决方案2】：

我通过conda install tensorflow-gpu 解决了这个问题。自动安装了cuDNN 7.3.1，问题解决了。

【讨论】：