torch==2.0.0
torchvision==0.15.1
pytorch_lightning==1.7.7
torchinfo==1.7.2
torchmetrics==0.9.3