From fa8f1c8db3cf00aa33e0e447567ea4c137ddda22 Mon Sep 17 00:00:00 2001 From: Spiros Maggioros Date: Thu, 12 Dec 2024 01:34:27 +0200 Subject: [PATCH] Correct torch version --- requirements.txt | 4 ++-- setup.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/requirements.txt b/requirements.txt index c1496e0..b53f4e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ -torch>=2.2.1 -nnunetv2>=2.2.1 +torch==2.2.1 +nnunetv2==2.5.1 argparse huggingface_hub pathlib diff --git a/setup.py b/setup.py index 384973f..25c52bc 100644 --- a/setup.py +++ b/setup.py @@ -20,9 +20,9 @@ download_url="https://github.com/CBICA/DLWMLS/", url="https://github.com/CBICA/DLWMLS/", packages=find_packages(exclude=["tests", ".github"]), - python_requires=">=3.8", + python_requires=">=3.9", install_requires=[ - "torch>=2.2.1", + "torch<=2.2.1", "nnunetv2>=2.2.1", "argparse", "huggingface_hub",