Error with pytorch_lightning for example training

[highfly-25]

I have setup a virtual environment in Google Colab with python 3.10 and installed the requirements "!pip install -r requirements/pt2.txt"

I am trying to run this command "python main.py --base configs/example_training/toy/cifar10_cond.yaml" in Google Colab but it throws the following error-
Traceback (most recent call last):
File "/content/gdrive/MyDrive/SDXL/generative-models/main.py", line 11, in
import pytorch_lightning as pl

• When I run "import pytorch_lightning as pl" separately then the error is thrown - ModuleNotFoundError: No module named 'pytorch_lightning'
• But as per the installed pip list result, the module is installed "!pip list | grep torch" :
  open_clip_torch 2.32.0
  pytorch-lightning 2.0.1
  torch 2.0.1
  torchaudio 2.0.2
  torchdata 0.6.1
  torchmetrics 1.7.0
  torchvision 0.15.2

I have setup the same code in Ubuntu server as well following the steps mentioned in Github but there also the same error is thrown.

Could you please help?

Thank you in advance.