torch>=1.4.0
torchvision>=0.4.1
pillow
