torch==2.2.0
torchvision==0.1.8
matplotlib==2.0.2
