torch==2.0.1
torchvision==0.15.2
pytorch-lightning==2.0.6
torchinfo==1.8.0
torchmetrics==1.0.1