torch
torchvision
numpy
matplotlib
pillow
tqdm