diff --git a/requirements.txt b/requirements.txt index cc0d597..6a6be03 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,5 +10,5 @@ scikit-learn>=0.24.0, !=0.24.1; python_version == '3.9' scikit-learn>=0.25.0; python_version == '3.10' scipy>=1.7 setuptools>=65.5.1 -torch>=1.7 +torch>=2.2.0 wheel>=0.38.0