torch>=1.6.0 numpy scikit-learn pyyaml pillow