Install torch cpu-only in github actions (grok-ai#91)

* Install torch cpu-only in github actions * Fix extra whitespace for pre-commits
brgsk · Aug 30, 2023 · e0d0bf6 · e0d0bf6
1 parent 669456e
commit e0d0bf6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/{{ cookiecutter.repository_name }}/.github/workflows/test_suite.yml b/{{ cookiecutter.repository_name }}/.github/workflows/test_suite.yml
@@ -61,6 +61,13 @@ jobs:
           sed -Ei '/^\s*-?\s*python\s*([#=].*)?$/d' ${{ env.CONDA_ENV_FILE }}
           cat  ${{ env.CONDA_ENV_FILE }}
 
+      # Install torch cpu-only
+      - name: Install torch cpu only
+        shell: bash -l {0}
+        run: |
+          sed -i '/nvidia\|cuda/d' ${{ env.CONDA_ENV_FILE }}
+          cat  ${{ env.CONDA_ENV_FILE }}
+
       - name: Setup Mambaforge
         uses: conda-incubator/setup-miniconda@v2
         with: