torch>=1.1.0
torchvison>=0.3.0
numpy
