torch>=1.0.0
numpy
tqdm
