python - Google Colab 错误 : Failed to get convolution algorithm. 这可能是因为 cuDNN 初始化失败

Question

UnknownError:获取卷积算法失败。这可能是因为 cuDNN 初始化失败，所以尝试查看上面是否打印了警告日志消息。

 [[{{node conv2d_1/convolution}} = Conv2D[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="VALID", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](zero_padding2d_1/Pad, conv2d_1/kernel/read)]]
 [[{{node metrics/acc/Mean/_255}} = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_563_metrics/acc/Mean", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]

这个错误是我在 Google colab 上遇到的。 Colab 的 tensorflow 版本是 1.12.0。

我找不到任何解决方案。因为都是本地系统。

Answer 1

对于 future 的路人来说，这看起来像是 Colab 人员现在正在解决的服务错误。

详情在the GitHub issue .

更新:此问题已在服务端修复。详情:https://github.com/googlecolab/colabtools/issues/384#issuecomment-451001849