torchvision>=0.5.0
matplotlib>=3.0.0
numpy>=1.16.0
tqdm>=4.20.0

pillow
sklearn