CancelledError 这错误难道跟显卡有关?

来源:7-12 subword文本分类之dataset变换与模型训练

AY9717

2019-12-28

我在 fit的时候,第一次成功了,然后第二次报出了这样的错误
CancelledError: [Derived]RecvAsync is cancelled.
[[{{node Adam/Adam/update/AssignSubVariableOp/_41}}]]
[[Reshape_11/_38]] [Op:__inference_distributed_function_6316]

Function call stack:
distributed_function

查了半天说是跟显卡有关,所以特来求证。

写回答

1回答

正十七

2020-01-02

对,貌似只跟GPU上的LSTM有关,参考:https://blog.csdn.net/qq_36659192/article/details/103198825

可以设置GPU auto growth来避免这个问题。

0
1
AY9717
非常感谢!
2020-01-03
共1条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程