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