Update pytorch to 2.1.2 + fix CI/linux CPU pytorch (#2429)

* Update pytorch to 2.1.2 * Change extra index url for cpu
chaiNNer-org · Jan 2, 2024 · d9c9361 · d9c9361
1 parent e67d839
commit d9c9361
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/backend/src/packages/chaiNNer_pytorch/__init__.py b/backend/src/packages/chaiNNer_pytorch/__init__.py
@@ -30,13 +30,13 @@ def get_pytorch():
             Dependency(
                 display_name="PyTorch",
                 pypi_name="torch",
-                version="2.1.1",
+                version="2.1.2",
                 size_estimate=55.8 * MB,
             ),
             Dependency(
                 display_name="TorchVision",
                 pypi_name="torchvision",
-                version="0.16.1",
+                version="0.16.2",
                 size_estimate=1.3 * MB,
             ),
         ]
@@ -45,23 +45,23 @@ def get_pytorch():
             Dependency(
                 display_name="PyTorch",
                 pypi_name="torch",
-                version="2.1.1+cu121" if nvidia_is_available else "2.1.1",
+                version="2.1.2+cu121" if nvidia_is_available else "2.1.2",
                 size_estimate=2 * GB if nvidia_is_available else 140 * MB,
                 extra_index_url=(
                     "https://download.pytorch.org/whl/cu121"
                     if nvidia_is_available
-                    else None
+                    else "https://download.pytorch.org/whl/cpu"
                 ),
             ),
             Dependency(
                 display_name="TorchVision",
                 pypi_name="torchvision",
-                version="0.16.1+cu121" if nvidia_is_available else "0.16.1",
+                version="0.16.2+cu121" if nvidia_is_available else "0.16.2",
                 size_estimate=2 * MB if nvidia_is_available else 800 * KB,
                 extra_index_url=(
                     "https://download.pytorch.org/whl/cu121"
                     if nvidia_is_available
-                    else None
+                    else "https://download.pytorch.org/whl/cpu"
                 ),
             ),
         ]