torch
numpy
torchvision
tqdm
h5py