matplotlib
numpy
tensorboard
torch
torchvision
tqdm

