TF2.0使用LSTM遇到 'CancelledError: [Derived]RecvAsync is cancelled.'问题解决

问题描述：

最近在学习TF2.0，使用普通的RNN可以正常训练，但是使用LSTM搭建网络，在fit时候遇到了如下错误：

CancelledError: [Derived]RecvAsync is cancelled.
[[{
  {node Reshape_17/_52}}]]
[[GroupCrossDeviceControlEdges_0/RMSprop/RMSprop/Const/_57]] [Op:__inference_distributed_function_24912]

Function call stack:
distributed_function

主机环境配置如下：

• 操作系统：Ubuntu18.04
• Tensorflow  version： 2.0.0
• Python version：3.6.7
• Cuda version：v10.0
• Cudnn version：v7.6.5
• GPU model and memory: GTX 1660ti  6GB

解决方法：

搜索问题发现很多人也遇到了这个问题，解决的办法有更新Cudnn版本等但没有解决，最后