matplotlib==3.8.0 Pillow==10.0.1 torch==2.3.1 torchsummary==1.5.1 torchvision==0.18.1