torch>=1.3.0
torchvision>=0.4.1
numpy
tqdm
packaging
