torch>=1.6.0
matplotlib
torchvision>=0.7.0
