torch
numpy
scipy
h5py
matplotlib
numba
ipython
imageio
colorlog
tqdm
scikit-learn
scikit-image
tensorboardX
torchvision
