tensorflow2.5可以正常运行但是2.0就报错

在GitHub上找了一个代码,代码要求使用tensorflow2.5,但是我的CUDA为10.1,不满足tensorflow2.5 gpu训练,于是使用tensorflow2.0
使用tensorflow2.5训练时可以正常训练(CPU训练),但是改为tensorflow2.0训练就报错。

你先测是下cuda10.1下能不能用tf2.5,不一定就用不了的,很多时候只是建议用而已。

1.不行的话升级下cuda

2.找下tf2.0下的接口用的是什么,手动改下这部分内容

   AttributeError: 'SepConvLSTM2D' object has no attribute '_maybe_reset_cell_dropout_mask'