The sample code that uses TPU gives an error.

[chizuchizu]

https://colab.research.google.com/github/PytorchLightning/pytorch-lightning/blob/master/notebooks/06-mnist-tpu-training.ipynb#scrollTo=YKt0KZkOUxVY

I used the sample code above as a reference and ran the same code in Kaggle Notebooks.
Below is a link to the Kaggle Notebooks with the code and the results.

https://www.kaggle.com/chizuchizu/minist-tpu-training

How can I make PyTorch Lightning work with TPU in Kaggle Notebooks?

/opt/conda/lib/python3.7/site-packages/torchvision/datasets/mnist.py:479: UserWarning: The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors. This means you can write to the underlying (supposedly non-writeable) NumPy array using the tensor. You may want to copy the array to protect its data or make it writeable before converting it to a tensor. This type of warning will be suppressed for the rest of this program. (Triggered internally at  /pytorch/torch/csrc/utils/tensor_numpy.cpp:179.)
  return torch.from_numpy(parsed.astype(m[2], copy=False)).view(*s)