torch>=0.4
numpy
tensorboardx
matplotlib
h5py
