MisconfigurationException: PyTorch XLA not installed

I am using below packages to install 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

related: https://github.com/PyTorchLightning/pytorch-lightning/issues/4829