CuDNNLSTM Kerasセルを使用してリカレントニューラルネットワーク(doc here)のトレーニング速度を向上させようとしています。Keras - ImportError:名前 'CuDNNLSTM'をインポートできません
私が実行します。
from keras.layers import Bidirectional, CuDNNLSTM
私はこのエラーを取得する:
ImportError: cannot import name 'CuDNNLSTM'
私の構成はKeras 2.0.8、Pythonの3.5、tensorflow-GPU 1.4.0(すべてのアナコンダによって管理される)でありますCUDA 8.0とcudnn 6.0の両方がインストールされているので、テンソルフローのnvidia依存性(here)に問題はありません。私のコード設定では、Kerasはテンソルフローのバックエンドを効果的に使用し、CuDNN *で始まるもの以外のすべてのレイヤーはうまく動作します。
誰かがこのインポートエラーの原因について考えていますか?