torch>=1.9.0 numpy>=1.21.0 scikit-learn>=1.0.0 Pillow>=8.0.0