PyTorch / XLA 可以用于 Coral 开发板 TPU 吗？

Question

我发现的最后一次讨论是在[本文]之前https://cloud.google.com/blog/products/ai-machine-learning/pytorch-is-now-ga-on-google-cloud-tpus 出来了。

有谁有任何经验或知道是否可以直接在开发板上使用 XLA 直接使用 pytorch 模型，而不是先转换为 TFLite 模型？

Answer 1

要在 Coral TPU 模型上运行模型，需要使用 compiler 进行编译，它将操作映射到 TPU/CPU。由于编译器仅接受 TFLite 模型作为输入，因此无法直接在 devboard 上使用 pytorch 模型。