Tensorflow / Keras：未知：无法获得卷积算法

Question

我具有以下硬件/软件：

• NVIDIA RTX2060 6GB
• Cuda 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。