torch
torchvision
Pillow

