numpy
torch>=1.13
torchvision>=0.14
