torch>=2.0.1
matplotlib
numpy
tqdm
