numpy<2.0.0,>=1.18.0
torch<2.5.0,>=1.6.0
torchvision<1.0.0,>=0.7.0
