torch>=0.4.0
torchvision>=0.1.9
tqdm>=4.19.4
matplotlib
numpy