We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3c25a57 commit 36edd65Copy full SHA for 36edd65
1 file changed
.github/workflows/nvidia-smoke.yml
@@ -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
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