matplotlib==3.8.3 numpy==1.26.4 torch==2.2.1 torchvision==0.17.1