Keras - 导入错误 : cannot import name 'CuDNNLSTM'

Question

我正在尝试使用 CuDNNLSTM Keras 单元来提高循环神经网络的训练速度(文档 here)。

当我运行时:

from keras.layers import Bidirectional, CuDNNLSTM

我收到此错误:

ImportError: cannot import name 'CuDNNLSTM'

我的配置是 Keras 2.0.8、python 3.5、tensorflow-gpu 1.4.0(均由 Anaconda 管理)并且我安装了 CUDA 8.0 和 cudnn 6.0，这对于 tensorflow 的 nvidia 依赖项应该没问题( here )。我的代码设置使 Keras 有效地使用了 tensorflow 后端，并且除了以 CuDNN* 开头的层之外的每一层都可以正常工作。

任何人都知道此导入错误的来源？

Answer 1

对于 Tensorflow-2:您可以只使用没有激活功能的 LSTM，它会自动使用 CuDNN 版本