diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 120ac902..83dc1835 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -22,7 +22,7 @@ jobs:
         run: |
           poetry install --with test
           poetry run python -m pip install --upgrade pip
-          poetry run pip install torch==2.0.1+cpu --index-url https://download.pytorch.org/whl/cpu
+          poetry run pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
       - name: Test with pytest
         run: |
           poetry run pytest tests/