torch==2.2.1
torchvision==0.17.1
matplotlib==3.7.1
numpy==1.25.2