change to cpu build torch

alan-turing-institute · Nov 8, 2023 · 649b0a7 · 649b0a7
1 parent 4180b82
commit 649b0a7
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -34,9 +34,6 @@ jobs:
           restore-keys: |
             ${{ runner.os }}-poetry-
       
-      - name: Install CPU-only version of PyTorch
-        run: pip3 install torch  --extra-index-url https://download.pytorch.org/whl/cpu
-
       - name: Install poetry
         run: |
           curl -sSL https://install.python-poetry.org | python -
@@ -47,6 +44,7 @@ jobs:
 
       - name: Install dependencies
         run: |
+          poetry run pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
           poetry install
 
       - name: Run Tests with Coverage