torch>=1.5.0
matplotlib
numpy
tqdm
scipy
h5py
