Install pytorch on github CI.

robohouse-delft · Jul 31, 2023 · b875657 · b875657
1 parent 9c0261b
commit b875657
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
@@ -8,6 +8,9 @@ jobs:
     name: Build and test
     runs-on: ubuntu-latest
     steps:
+      - name: Install pytorch
+        run: pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
       - name: Checkout code
         uses: actions/checkout@master
 
@@ -25,15 +28,18 @@ jobs:
         with:
           command: build
           args: --release --features full --color=always
+          env: LIBTORCH_USE_PYTORCH=1
 
       - name: Clippy
         uses: actions-rs/clippy-check@v1
         with:
           token: ${{ secrets.GITHUB_TOKEN }}
           args: --release --features full
+          env: LIBTORCH_USE_PYTORCH=1
 
       - name: Test
         uses: actions-rs/cargo@v1
         with:
           command: test
           args: --release --features full --color=always
+          env: LIBTORCH_USE_PYTORCH=1