module 'tensorflow.python.keras.api._v2.keras.layers' has no attribute 'CuDNNLSTM'

Question

When I write tf.keras.layers.LSTM, I get the warning

Note that this layer is not optimized for performance. Please use tf.keras.layers.CuDNNLSTM for better performance on GPU.

But when I change the layer to tf.keras.layers.CuDNNLSTM, I get the error

AttributeError: module 'tensorflow.python.keras.api._v2.keras.layers' has no attribute 'CuDNNLSTM'

Tensorflow version is 2.0.0-alpha0, Keras version is 2.2.4-tf.

How can I fix this problem?

There is no CuDNNLSTM layer in tf 2.0 yet. See the available layers here tensorflow.org/versions/r2.0/api_docs/python/tf/keras/layers — Sreeram TP
– Sreeram TP, Commented Apr 19, 2019 at 12:06
Use tf.compat.v1.keras.layers.CuDNNLSTM(), not tf.keras.layers.CuDNNLSTM() — Vlad
– Vlad, Commented Apr 19, 2019 at 12:24

Stewart_R · Accepted Answer · 2019-07-01 05:20:34Z

13

In general, in TensorFlow 2.0 we should just use:

tf.keras.layers.LSTM

which, despite the warning, will use the GPU.

The warning message incorrectly existed in the 2.0.0-alpha0 version but has since been removed in 2.0.0-beta1

If for some reason you specifically need the original implementation of tf.keras.layers.CuDNNLSTM then you can use tf.compat.v1.keras.layers.CuDNNLSTM but this would be an edge case.

edited Jul 1, 2019 at 5:20

answered Jun 26, 2019 at 11:53

Stewart_R

14.5k11 gold badges66 silver badges118 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

module 'tensorflow.python.keras.api._v2.keras.layers' has no attribute 'CuDNNLSTM'

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related