torch>=1.9
torchvision>=0.10.0