Minimum of torch 2.2.0 during build (#649)

casper-hansen · Nov 16, 2024 · 8f71dc8 · 8f71dc8
1 parent 887d08b
commit 8f71dc8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -44,6 +44,8 @@ jobs:
     defaults:
       run:
         shell: pwsh
+    env:
+      MIN_TORCH_VER: "2.2.0"
 
     steps:
       - name: Free Disk Space
@@ -78,7 +80,7 @@ jobs:
       - name: Install Dependencies
         run: |
           # Install torch
-          python -m pip install --upgrade --no-cache-dir torch==2.5.1
+          python -m pip install --upgrade --no-cache-dir torch==$env:MIN_TORCH_VER
           python -m pip install build setuptools wheel
 
           # Print version information