numpy
torch
torchvision
matplotlib
tensorboard
scipy
scikit-learn
tqdm
h5py