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

Question

我写的时候tf.keras.layers.LSTM得到警告

Note that this layer is not optimized for performance. Please use tf.keras.layers.CuDNNLSTM for better performance on GPU.

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

AttributeError: module 'tensorflow.python.keras.api._v2.keras.layers' has no attribute '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但这将是一个边缘情况。