numpy<3,>=2
torch<3,>=2
torchvision<1,>=0.17
