scikit-learn==1.2.2 torch==1.13.1 torchvision==0.14.1