python - tensorflow /Keras : Unknown: Failed to get convolution algorithm

Question

我有以下硬件/软件:

• NVIDIA RTX2060 6GB
• 库达10

我想运行Keras mnist example 。在我的另一台计算机上，可以使用相同的软件。我收到此错误:

tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: 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.
         [[{{node conv2d_1/convolution}}]]
         [[conv2d_19/Sigmoid/_1405]]
  (1) Unknown: 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.
         [[{{node conv2d_1/convolution}}]]

我的问题出在哪里？

Answer 1

此错误通常是由于 CUDA/cudnn 版本不匹配造成的。请按照提供的链接中的说明进行操作:https://www.tensorflow.org/install/gpu 。另外，如果你的默认 python 来自 Anaconda，你应该使用 conda 安装 evrything。