我写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。
如何解决此问题？
答案 0 :(得分：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，但这是一个极端的情况。