torch
torchvision
gdown
numpy
scikit-learn