Returned torch 1.11 version as newer ones are not comptible with segm…

…odelspytorch
InnopolisUni · Sep 12, 2024 · 30b98da · 30b98da
1 parent b246526
commit 30b98da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pckg_util.py b/pckg_util.py
@@ -63,13 +63,13 @@ def check_gpu_and_torch_compatibility():
             if "NVIDIA A100" in output:
                 install_and_import(
                     "torch",
-                    "1.12.0+cu113",
+                    "1.11.0+cu113",
                     "-f",
                     "https://download.pytorch.org/whl/torch_stable.html",
                 )
                 install_and_import(
                     "torchvision",
-                    "0.13.0+cu113",
+                    "0.12.0+cu113",
                     "-f",
                     "https://download.pytorch.org/whl/torch_stable.html",
                 )

diff --git a/pyproject.toml b/pyproject.toml
@@ -15,8 +15,8 @@ python-dotenv = "^0.20.0"
 
 
 
-torch = {version = "1.12.0", source = "cudatorch"}
-torchvision = {version = "0.13.0", source = "cudatorch"}
+torch = {version = "1.11.0", source = "cudatorch"}
+torchvision = {version = "0.12.0", source = "cudatorch"}
 
 
 tqdm = "^4.64.0"