Definition of setup versions

ayrna · Nov 20, 2023 · 34f358a · 34f358a
1 parent dee720d
commit 34f358a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -11,8 +11,8 @@
     install_requires=[
         "scikit-learn",
         "numpy",
-        "torch",
-        "torchvision",
+        "torch==1.10.2",
+        "torchvision==0.11.3",
         "pandas",
         "scipy",
         "matplotlib",