在PyTorch中使用TPU

Question

我正在尝试使用Colab的Google Cloud TPU.我可以通过使用Tensorflow来完成本教程.

有人知道是否可以使用PyTorch来使用TPU？如果是这样我该怎么办？你有什么例子吗？

Answer 1

目前,由于专为Tensorflow设计,因此不可能将Cloud TPU与PyTorch一起使用.

但是,根据三天前谷歌云博客发布的产品新闻,"谷歌TPU团队的工程师正积极与PyTorch核心开发人员合作,将PyTorch连接到云端TPU".

Answer 2

查看我们的存储库pytorch/xla，您可以在其中开始在 TPU 上训练 PyTorch 模型。

此外，您甚至可以通过这些Colab 笔记本在带有 PyTorch 的 Colab 上使用免费的 TPU 。