2017-12-17 29 views
2

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 *で始まるもの以外のすべてのレイヤーはうまく動作します。

誰かがこのインポートエラーの原因について考えていますか?

答えて

2

ケア2.0.8には、最新のバージョンに付属しているこれらの種類の層のコードがありません。

私は、最新のバージョンにアップグレードするPIPを使用:

pip install --upgrade keras 

とそれはすべてが動作するようになりました。

関連する問題