您怎么知道您是否在TensorFlow 2.0中使用LSTM或GRU的CuDNN版本？

Question

TensorFlow 2.0的一个不错的功能：new LSTM layer现在自动决定是否使用CuDNN实现（GRU层也是如此）。但是它们的使用不仅取决于GPU，还取决于它。还有其他一些参数需要设置（例如unroll=False）。

我正在使用带有GPU（不是非常强大的GPU）的tensorflow-gpu，而我的RNN像糖蜜一样慢。我想确定我正在运行CuDNN实现。日志中有什么地方可以查看吗？

我也用过tf.compat.v1.keras.layers.CuDNNLSTM，对我来说也一样慢。我可以确定这是使用CuDNN实现，还是在某些情况下可以退回到其他东西？

Answer 1

如果使用tf.compat.v1.keras.layers.CuDNNLSTM，则肯定是使用cuDNN实现。那里没有退缩。