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"

你可能感兴趣的:(bugs,deep,learning,tensorflow,python,lstm)