torch>=2.0.0
tqdm
numpy
matplotlib
