函数调用堆栈_dummy_computation_fn，在训练我的对象检测模型时，我会收到这个错误

Question

错误可能起源于输入操作。连接到节点ResNet101V1_FPN/model/conv/conv2D的输入源操作: ResNet101V1_FPN/model/lambda/Pad (定义在/local/lib/python3.7/dist-packages/object_detection/models/keras_models/resnet_v1.py:51)上)

函数调用堆栈：_dummy_computation_fn

Answer 1

当我遇到这个问题时，我正在使用Tensorflow-gpu=2.6.0。在寻找和看到许多建议之后。我相信Tf2.6.0没有与CuDNN 8兼容的版本，所以我将TF降级为2.4.1 !pip --升级tensorflow-gpu=2.4.1，它成功了。另一个建议是卸载TensorFlow和tensoflow，然后重新启动运行时，安装您想要的TensorFlow版本。

可能的解决办法：

1. 将Batch_size从64减少到8或4或更低。
2. 重新启动运行时并卸载所有Tensorflow-gpu版本。

我希望我的回答会有所帮助。