matplotlib
numpy
torch>=1.5.1
torchvision>=0.6.1
