可以将CuDNNLSTM与Google Colab的TPU一起使用吗？

Question

我可以使用他们的 GPU 来执行此操作，但是使用他们的 TPU 可以找回错误... 请问周围的人知道我我 丢失吗？ 实际将TPU与CuDNNLSTM一起使用是否有意义？还是CuDNNLSTM只是为GPU量身定制的？ 提前谢谢。

Answer 1

keras.layers.CuDNNLSTM仅在GPU上受支持。但是在Tensorflow 2中，内置的LSTM和GRU层已更新为在有GPU时默认使用CuDNN内核。

以下是Performance optimization and CuDNN kernels中的详细信息：

在TensorFlow 2.0中，内置的LSTM和GRU层已更新，以在有GPU时默认使用CuDNN内核。 通过此更改，先前的keras.layers.CuDNNLSTM / CuDNNGRU层已被弃用，您可以构建模型而不必担心它将运行的硬件。

您可以只使用内置的LSTM层：tf.keras.layers.LSTM，它将同时在TPU和GPU上运行。