PyTorch TPU support on Google Colab

When can I train PyTorch models on Google Colab Cloud TPUs ? I’m trying to avoid shifting to tensorflow for my project just for the TPUs.

1 Like

I guess we’ll have to wait a really long time…

You can try Google Colab by using following jupyter notebook.

From seeing following url, nightly build package for pytorch_xla exist.

Which combination with torch_xla & torch is latest?
for example, which torch version should be work with wheels/torch_xla-20190508-0.1+d581df3-cp35-cp35m-linux_x86_64.whl?