Tensorflow/Keras：未知：無法獲得卷積算法

Question

我有以下硬件/軟件：

• 英偉達 RTX2060 6GB
• Cuda 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 的 vesion 不匹配而發生的。 請按照提供的鏈接上的說明進行操作： https://www.tensorflow.org/install/gpu 。 此外，如果您的默認 python 來自 Anaconda，您應該使用 conda 安裝 evrything。