torch>=1.1
torchvision>=0.3.0
numpy
