python - 模块 'tensorflow.python.keras.api._v2.keras.layers' 没有属性 'CuDNNLSTM'

Question

当我编写 tf.keras.layers.LSTM 时，我收到了警告

请注意，该层并未针对性能进行优化。请使用 tf.keras.layers.CuDNNLSTM 以获得更好的 GPU 性能。

但是当我将图层更改为 tf.keras.layers.CuDNNLSTM 时，出现错误

AttributeError: 模块 'tensorflow.python.keras.api._v2.keras.layers' 没有属性 'CuDNNLSTM'

Tensorflow 版本为 2.0.0-alpha0，Keras 版本为 2.2.4-tf。

我该如何解决这个问题？

Answer 1

一般来说，在 TensorFlow 2.0 中我们应该只使用:

tf.keras.layers.LSTM

尽管有警告，将使用 GPU。

警告消息错误地存在于 2.0.0-alpha0 版本中，但已在 2.0.0-beta1 中删除

如果出于某种原因您特别需要 tf.keras.layers.CuDNNLSTM 的原始实现，那么您可以使用 tf.compat.v1.keras.layers.CuDNNLSTM但这将是一个边缘案例。