numpy
scipy
scikit-learn
torch>=0.4.0
matplotlib
torchvision
