torch
torchvision
h5py
matplotlib
pandas
