Example on training with TPU does not run at all
I am currently try this colab notebook https://colab.research.google.com/github/PytorchLightning/lightning-tutorials/blob/publication/.notebooks/lightning_examples/mnist-tpu-training.ipynb#scrollTo=2772a2e1 provided by PL teams to get some experience with TPU training. But when I try to execute the third cell, there is some Import error with the _XLAC module.
Hi @tungts1101! this error is raised when the PyTorch and PyTorch xla are not of the same versions.
You could verify using pip list | grep torch and could install the latest versions for both!