torch>=2.1.0
torchvision>=0.16.0
scikit-learn>=1.2.0
