Keras - ImportError：无法导入名称'CuDNNLSTM'

Question

我正在尝试使用CuDNNLSTM Keras单元来提高复发神经网络（doc here）的训练速度。

当我跑：

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（here）的nvidia依赖项。我的代码设置使得Keras有效地使用张量流后端，除了以CuDNN *开头的那些层之外的每个层都工作正常。

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

Answer 1

事实证明keras 2.0.8没有更新版本中出现的这类图层的代码。

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

pip install --upgrade keras

这一切现在都有效。