pin torch to 2.5.1 in pip & cpu ci/cd config

mllam · Jan 28, 2025 · 36a60e6 · 36a60e6
1 parent ca56005
commit 36a60e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/ci-pip-install-and-test-cpu.yml b/.github/workflows/ci-pip-install-and-test-cpu.yml
@@ -15,7 +15,10 @@ jobs:
 
       - name: Install torch (CPU)
         run: |
-          python -m pip install torch  --index-url https://download.pytorch.org/whl/cpu
+          # torch version pinned to 2.5.1 for now so that we avoid the
+          # downgrade (and then GPU version install) when actually installing
+          # neural-lam
+          python -m pip install "torch==2.5.1"  --index-url https://download.pytorch.org/whl/cpu
 
       - name: Install package (including dev dependencies)
         run: |