tensorflow.python.framework.errors_impl.CancelledError: [_Derived_]RecvAsync is cancelled.

问题描述：

使用tensorflow-gpu时，跑lstm，报了以下错误。具体原因不明，有知道的朋友帮我解释一下吧，谢谢！ 参考：link

• 环境

python == 3.6
tensorflow-gpu == 2.4.0

• 报错复现

tensorflow.python.framework.errors_impl.CancelledError: [_Derived_]RecvAsync is cancelled. [[{{node Adam/Adam/group_deps/NoOp/_95}}]] [[gradient_tape/model/embedding_1/embedding_lookup/Reshape/_90]] [Op:__inference_train_function_11140]

解决方案：

在train之前加一句：

os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"