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: "MisconfigurationException: PyTorch XLA not installed." Even though PyTorch XLA is installed · Issue #4829 · Lightning-AI/lightning · GitHub