Extend pip index with torch cpu

tenstorrent · Jun 25, 2024 · 4903446 · 4903446
1 parent 85319e8
commit 4903446
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/before_merge.yaml b/.github/workflows/before_merge.yaml
@@ -25,7 +25,8 @@ jobs:
         run: |
           python3 -m venv venv
           source venv/bin/activate
-          python3 -m pip install --upgrade pip
+          python3 -m pip config set global.extra-index-url https://download.pytorch.org/whl/cpu
+          python3 -m pip install --upgrade pip          
           python3 -m pip install -r requirements.txt      
       - name: Run Tests
         run: |