CancelledError: [_Derived_]RecvAsync is cancelled.[[{{node Adam/Adam/update/AssignSubVariab...解决方案

程序员文章站 2024-03-11 17:07:01

...

使用tensorflow2做LSTM时报“CancelledError: [Derived]RecvAsync is cancelled.
[[{{node Adam/Adam/update/AssignSubVariableOp/_43}}]]
[[gradient_tape/sequential_2/embedding_2/embedding_lookup/Reshape/_40]] [Op:__inference_train_function_12331]”

这个是显存的问题，显卡太垃圾了导致默认分配的显存不够。
在导入模块开始时，添加如下代码，开启动态使用显存方式

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