python==3.10.*
torch==2.2.*
torchvision==0.17.*
numpy==1.21.*
scikit-learn==1.0.*