Revert torch cpu install

atong01 · Aug 30, 2023 · 3cc83c0 · 3cc83c0
1 parent aacbec5
commit 3cc83c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -32,7 +32,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           # Install cpu versions
-          pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          # pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install -r requirements.txt
           pip install pytest
           pip install sh
@@ -62,7 +62,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          # pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install -r requirements.txt
           pip install pytest
           pip install pytest-cov[toml]