使用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"