torch
torchvision
Pillow
matplotlib
