numpy<2,>=1.20
torch<2,>=1.9
torchvision<1,>=0.10
