matplotlib==3.4.3
numpy==1.21.1
torch==1.9.0
torchvision==0.10.0