diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
index b28b9b9..43af334 100644
--- 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