numpy
pillow
pytorch>=0.4
h5py
