Can't install torch==2.2.1 with nightly version, on hold till they fu…

…lly support it
CBICA · Nov 9, 2024 · 4647f05 · 4647f05
1 parent fcc196c
commit 4647f05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
     install_requires=[
-        "torch",
+        "torch<=2.2.1",
         "nnunetv2<=2.2.1",
         "argparse",
         "huggingface_hub"