tensorflow - 如何在本地 rtx gpu 的 tpu 上训练 keras 模型

Question

我使用 keras 后端张量流。

我的本地电脑上有 RTX gpu，我想在 tpu（张量核心）上训练 keras 模型

有什么方法吗？

Answer 1

RTX GPU 和 tpu og google 的 Tensorcore 是不同的。