torch>=2.1.2
torchvision>=0.16.2
