Google Colab 错误:获取卷积算法失败。这可能是由于cuDNN未能初始化。

8

未知错误:获取卷积算法失败。这可能是因为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。
我找不到任何解决方法,因为所有的解决方案都是针对本地系统的。

你能分享一个自包含的笔记本,以重现你观察到的错误吗? - Bob Smith
1
你解决了这个问题吗?如果解决了,你是怎么解决的? - Jonathan
我收到了相同的错误。似乎只有在你分配了GPU时才会出现这个错误。当我使用CPU训练模型时,我没有遇到这个错误。 - Sam Perry
2个回答

1

2
还没有解决。出现了这个讨厌的错误。 - Nguai al

0

只需安装tensorflow,不需要安装CUDA和cuDNN

这是我的第一段代码(出现错误的代码)

!pip卸载tensorflow

!pip安装tensorflow-gpu==1.15

!apt安装--allow-change-held-packages libcudnn7=7.4.1.5-1+cuda10.0

通过删除最后一行,我的错误得到了解决

!pip卸载tensorflow

!pip安装tensorflow-gpu==1.15

希望能帮到你!


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接