torch>=1.6
torchvision>=0.3.0
numpy
