matplotlib
seaborn
numpy
tqdm
h5py
gitpython
torch

[pytorch]
torch
torch-cuda
