Tensorflow/Keras:未知:无法获取卷积算法

Question

我有以下硬件/软件：

• 英伟达 RTX2060 6GB
• 库达 10

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

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。