google-cloud-platform - PyTorch での TPU の使用

Question

Colab の Google Cloud の TPU を使用しようとしています。Tensorflowを使用して、チュートリアルに従って実行できました。

PyTorch を使用して TPU を利用できるかどうか知っている人はいますか? もしそうなら、どうすればできますか？例はありますか？

Answer 1

Cloud TPU は Tensorflow 専用に設計されているため、現在、PyTorch で Cloud TPU を使用することはできません。

しかし、Google Cloud ブログに 3 日前に投稿されたこの製品ニュースによると、「Google の TPU チームのエンジニアは、コアの PyTorch 開発者と積極的に協力して、PyTorch を Cloud TPU に接続しています」。