torch>=1.9.0
torchvision
pytorch-lightning
Pillow
numpy
scipy
matplotlib
scikit-learn
h5py
