I am trying to train my model on tpu on kaggle.
I used the follwoing code for installing torch_xla
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
!python pytorch-xla-env-setup.py --version nightly --apt-packages libomp5 libopenblas-dev

But while importing I am getting error
import torch_xla.core.xla_model as xm
import torch_xla.distributed.parallel_loader as pl
import torch_xla.distributed.xla_multiprocessing as xmp


image1337×377 61.7 KB

Can somebody help me understand that why I am getting this error and how can I make it work