Install torch CPU for workflows

kddubey · May 30, 2024 · e04506a · e04506a
1 parent 781bb8b
commit e04506a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -26,6 +26,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
+          python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           python -m pip install .[dev]
 
       - name: Run tests

diff --git a/.readthedocs.yaml b/.readthedocs.yaml
@@ -14,4 +14,8 @@ python:
     - method: pip
       path: .
       extra_requirements:
+        - torch
+        - torchvision
+        - torchaudio
+        - --extra-index-url https://download.pytorch.org/whl/cpu
         - dev