matplotlib==3.4.3
numpy==1.21.5
torch==1.10.1
torchvision