模块“tensorflow.python.keras.api._v2.keras.layers”没有属性“CuDNNLSTM”答案

【问题标题】：module 'tensorflow.python.keras.api._v2.keras.layers' has no attribute 'CuDNNLSTM'模块“tensorflow.python.keras.api._v2.keras.layers”没有属性“CuDNNLSTM”
【发布时间】：2019-09-09 16:40:27
【问题描述】：

当我写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。

我该如何解决这个问题？

【问题讨论】：

tf 2.0 中还没有CuDNNLSTM 层。在此处查看可用层tensorflow.org/versions/r2.0/api_docs/python/tf/keras/layers
使用tf.compat.v1.keras.layers.CuDNNLSTM()，而不是tf.keras.layers.CuDNNLSTM()

标签： python tensorflow keras deep-learning

【解决方案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，但这将是一种极端情况。

【讨论】：