torch==2.0.1 
torchvision==0.15.2 
torchaudio==2.0.2
lightning==2.0.0
omegaconf==2.3.0
ninja