Skip to content

Commit 36edd65

Browse files
committed
Smoke test
1 parent 3c25a57 commit 36edd65

1 file changed

Lines changed: 24 additions & 0 deletions

File tree

.github/workflows/nvidia-smoke.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
name: gpu-smoke
2+
on:
3+
workflow_dispatch: {}
4+
push:
5+
branches: [ main ]
6+
7+
jobs:
8+
smoke:
9+
runs-on: [self-hosted, gpu]
10+
timeout-minutes: 30
11+
steps:
12+
- uses: actions/checkout@v4
13+
- name: Show GPU
14+
run: |
15+
nvidia-smi || true
16+
- name: Torch sanity
17+
run: |
18+
python - <<'PY'
19+
import torch
20+
print("CUDA available:", torch.cuda.is_available())
21+
print("device_count:", torch.cuda.device_count())
22+
if torch.cuda.is_available():
23+
print("device_0:", torch.cuda.get_device_name(0))
24+
PY

0 commit comments

Comments
 (0)