install pytorch cpu build if no nvidia-smi command

henrysky · Sep 3, 2024 · f58abb3 · f58abb3
1 parent b0d5666
commit f58abb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci_tests.yml b/.github/workflows/ci_tests.yml
@@ -32,7 +32,7 @@ jobs:
           sudo apt install graphviz
           pip install -r requirements.txt
           pip install -r requirements-dev.txt
-          pip install ${{ matrix.BACKEND }}~=${{ matrix.BACKEND_VER }}
+          if [ -x "$(command -v nvidia-smi)" ] || [ "${BACKEND}" == "tensorflow" ]; then pip install ${{ matrix.BACKEND }}~=${{ matrix.BACKEND_VER }}; else pip install torch~=${{ matrix.BACKEND_VER }} --index-url https://download.pytorch.org/whl/cpu; fi
           pip install .
       - name: Cache test data
         uses: actions/cache@v4