numpy>=1.14.2 torch==0.4.0 torchvision>=0.1.8