torch
torchvision
numpy
scikit-learn