Skip to content

Commit 7fc6107

Browse files
committed
Update with GPU runner
Signed-off-by: Eric Kerfoot <[email protected]>
1 parent d7e1451 commit 7fc6107

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/cicd_tests.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ jobs:
5959
strategy:
6060
fail-fast: false
6161
matrix:
62-
os: [windows-latest, macOS-latest, ubuntu-latest]
62+
os: [windows-latest, macOS-latest, ubuntu-latest, linux-gpu-runner]
6363
python-version: ['3.9']
6464
pytorch-version: ['2.5.1']
6565
include:
@@ -110,6 +110,7 @@ jobs:
110110
key: ${{ matrix.os }}-latest-pip-${{ steps.pip-cache.outputs.datew }}
111111
- name: Install the dependencies with PyTorch ${{ matrix.pytorch-version }}
112112
run: |
113+
/usr/bin/env nvidia-smi &> /dev/null || echo "No nvidia-smi found"
113114
# min. requirements
114115
python -m pip install torch==${{ matrix.pytorch-version }}
115116
python -m pip install -r requirements-min.txt

0 commit comments

Comments
 (0)