torch>=1.5.1
torchvision>=0.6.1
matplotlib
numpy
scipy
