torch>=1.10
torchvision>=0.11
pillow
matplotlib
