torch>=1.0
torchvision
shutil
