How to run Pytorch code on Kaggle's TPU's

Hi All,

I have some code which I borrowed from here. As seen in the tutorial, it runs on the GPU. Would anyone know how to convert it to run on Kaggle’s TPU’s please.

Thanks & Best Regards

Michael Schtoter