pillow
torch
torchvision
matplotlib
accelerate