torch>=1.9
matplotlib
tqdm
