Merge pull request #43 from robohouse-delft/fix-ci

Bump torch version installed in CI.
robohouse-delft · Mar 30, 2024 · 9598b06 · 9598b06
2 parents 722afd7 + 7f4f598
commit 9598b06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
@@ -10,7 +10,7 @@ jobs:
     steps:
       - name: Install pytorch
         run: |
-          pip3 install torch==2.0.0 torchvision==0.15.0 --index-url https://download.pytorch.org/whl/cpu
+          pip3 install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu
           echo LIBTORCH_USE_PYTORCH=1 >> "$GITHUB_ENV"
           echo LD_LIBRARY_PATH="$(python -m site --user-site)/torch/lib${LD_LIBRARY_PATH:+:}$LD_LIBRARY_PATH" >> "$GITHUB_ENV"
       - name: Checkout code