torch
torchvision
setuptools
tqdm
numpy