h5py
hdf5storage
matplotlib
mpi4py
numpy
pillow
pyyaml
scipy
torch
torchvision
