matplotlib
seaborn
numpy
tqdm
gitpython
torch

[pytorch]
torch
torch-cuda
