Question

我正在使用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， Tensorflow版本：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]

有效！有人知道我为什么要遭受这个问题吗？

Answer 1

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

尝试检查您拥有的cudnn和tensorflow版本。

Answer 2

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

tensorflow cuDNN兼容性

2 个答案: