torch>=2.0.0
torchvision>=0.17.0
pytorch-lightning>=2.2.0
