Keras-ImportError：无法导入名称“ CuDNNLSTM”

Question

我试图使用CuDNNLSTM Keras细胞，以提高训练速度的回归神经网络（DOC 这里）。

当我跑步时：

from keras.layers import Bidirectional, CuDNNLSTM

我收到此错误：

ImportError：无法导入名称“ CuDNNLSTM”

我的配置是Keras 2.0.8，python 3.5，tensorflow-gpu 1.4.0（均由Anaconda管理），并且我同时安装了CUDA 8.0和cudnn 6.0，应该可以与tensorflow的nvidia依赖关系（在此处）配合使用。我的代码设置使Keras有效地使用了tensorflow后端，除以CuDNN *开头的层以外的其他所有层都可以正常工作。

有人知道此导入错误的来源吗？

Answer 1

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

Answer 2

事实证明，keras 2.0.8没有较新版本中提供的用于此类图层的代码。

我使用pip升级到最新版本：

pip install --upgrade keras

现在一切正常。