torch==2.0.0
torchvision
lightning==2.0.1
torchmetrics==1.0.0