torch==1.8.0
torchvision
Pillow