numpy==1.24.3 matplotlib==3.7.1 pillow==9.5.0 torch==1.8.0+cu111 torchvision==0.9.0+cu111