torch>=2.0.0
torchvision>=0.17.0
