Using TPUs with PyTorch

google-cloud-platform pytorch google-colaboratory tpu google-cloud-tpu
Salvador Medina picture Salvador Medina · Oct 4, 2018 · Viewed 7.9k times · Source

I am trying to use Google Cloud's TPU from Colab. I was able to do it following the tutorial by using Tensorflow.

Does anybody know if it is possible to make use of the TPUs using PyTorch? If so how can I do it? Do you have any example?

Answer

Héctor Neri picture Héctor Neri · Oct 5, 2018

Currently, it's not possible to use Cloud TPU with PyTorch since it's designed specifically for Tensorflow.

But, according to this product news posted three days ago in the Google Cloud blog, "engineers on Google’s TPU team are actively collaborating with core PyTorch developers to connect PyTorch to Cloud TPUs".

Related questions

Google server putty connect 'Disconnected: No supported authentication methods available (server sent: publickey)

I'm trying to connect to my Debian Google Compute Engine server through PuTTy (I've tried other alternatives too) but when I do I get the error "Disconnected: No supported authentication methods available (server sent: publickey) The google server came without …

Read More
How to open a specific port such as 9090 in Google Compute Engine

I have 2 Google Compute Engine instances and I want to open port 9090 in both the instances. I think we need to add some firewall rules. Can you tell me how can I do that?

Read More
How to change the project in GCP using CLI commands

How can i change the current running project to another project in GCP (Google Cloud Platform) account using cli commands other than using gcloud init manually. $gcloud projects list will list the projects running on my account. I want to …

Read More