numpy<2 matplotlib==3.9.4 torch==2.1.2 torchvision==0.16.2 clize==5.0.2 scikit-image==0.22.0