tensorflow.keras.layers：ImportError：无法导入名称“CuDNNLSTM”

Question

请我尝试从 tensorflow.keras.layers 导入 CuDnnLSTM 以提高我的训练速度，但出现此错误。我知道使用 Keras 2.0.8 和 python 3.5 的用户提出了类似的问题。

我的配置是 tensorflow 版本 2.0.0-beta1 和 Python 3.6.10。

这是我尝试过的： from tensorflow.keras.layers import CuDNNLSTM

我得到了这个错误， ImportError：无法导入名称“CuDNNLSTM”

请问有人知道如何解决此错误吗？提前致谢！

Answer 1

您可以使用from tensorflow.compat.v1.keras.layers import CuDNNLSTM，而不是from tensorflow.keras.layers import CuDNNLSTM。

请参考下图的工作代码

import tensorflow as tf
print(tf.__version__)
from tensorflow.keras.models import Sequential
from tensorflow.compat.v1.keras.layers import CuDNNLSTM

model = Sequential()
model.add(CuDNNLSTM(1, return_sequences=0, input_shape=(1, 1)))
print(model.summary())

输出：

2.0.0-beta1
Model: "sequential"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
cu_dnnlstm (CuDNNLSTM)       (None, 1)                 16        
=================================================================
Total params: 16
Trainable params: 16
Non-trainable params: 0
_________________________________________________________________
None