你怎么知道你在 TensorFlow 2.0 中使用的是 LSTM 或 GRU 的 CuDNN 版本？答案

【问题标题】：How do you know if you're using the CuDNN version of LSTM or GRU in TensorFlow 2.0?你怎么知道你在 TensorFlow 2.0 中使用的是 LSTM 或 GRU 的 CuDNN 版本？
【发布时间】：2019-11-22 21:08:01
【问题描述】：

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

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

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

【问题讨论】：

标签： tensorflow keras lstm recurrent-neural-network

【解决方案1】：

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

【讨论】：