"MisconfigurationException: PyTorch XLA not installed." Even though PyTorch XLA is installed
🐛 Bug

I'm using exactly the same command to install pytorch xla as shown in the doc:
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
However, I still got this error:
GPU available: False, used: False
TPU available: True, using: 1 TPU cores
training on 1 TPU cores
---------------------------------------------------------------------------
MisconfigurationException                 Traceback (most recent call last)
<ipython-input-11-1f9f6fbe4f6c> in <module>()
----> 1 test_x(tmpdir)

2 frames
/usr/local/lib/python3.6/dist-packages/pytorch_lightning/accelerators/tpu_accelerator.py in setup(self, model)
     61         # TODO: Move this check to Trainer __init__ or device parser
     62         if not TPU_AVAILABLE:
---> 63             raise MisconfigurationException('PyTorch XLA not installed.')
     64 
     65         # see: https://discuss.pytorch.org/t/segfault-with-multiprocessing-queue/81292/2

MisconfigurationException: PyTorch XLA not installed.

Please reproduce using the BoringModel and post here

https://colab.research.google.com/drive/1dz4NY233VoBv9SIb22JnrWiqYBHFpxCm?usp=sharing
To Reproduce

run through the colab notebook in the link above
Expected behavior
no error
Environment
Note: Bugs with code are solved faster ! Colab Notebook should be made public !


IDE: Please, use our python bug_report_model.py template.


Colab Notebook: Please copy and paste the output from our environment collection script (or fill out the checklist below manually).


You can get the script and run it with:
wget https://raw.githubusercontent.com/PyTorchLightning/pytorch-lightning/master/tests/collect_env_details.py
# For security purposes, please check the contents of collect_env_details.py before running it.
python collect_env_details.py


CUDA:

GPU:
available:         False
version:           None


Packages:

numpy:             1.18.5
pyTorch_debug:     False
pyTorch_version:   1.8.0a0+4ed7f36
pytorch-lightning: 1.0.7
tqdm:              4.41.1


System:

OS:                Linux
architecture:

64bit



processor:         x86_64
python:            3.6.9
version:           #1 SMP Thu Jul 23 08:00:38 PDT 2020



Additional context