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.
https://github.com/pytorch/xla/tree/master/contrib/colab
From seeing following url, nightly build package for pytorch_xla exist.
http://storage.googleapis.com/pytorch-tpu-releases/
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?