torch>=1.7.0
numpy
pandas
matplotlib
tqdm
