numpy
torch>=1.8.1+cu111
torchvision>=0.9.1+cu111
scipy
sklearn
matplotlib
h5py
