python==3.7.9
torch== 1.7.1+cu110
numpy
h5py
matplotlib

