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