Skip to content

Commit 6437e19

Browse files
committed
feat
1 parent fbc28ad commit 6437e19

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

.github/workflows/nvidia-arc-health.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,9 @@ jobs:
1212
runs-on: [nvidia-docker-b200-8-x86-64]
1313
timeout-minutes: 5
1414

15+
steps:
1516
- name: GPU Health Check
16-
run: python3 -c "import torch; torch.randn(5, device='cuda')"
17+
run: python -c "import torch; torch.randn(5, device='cuda')"
1718

1819
env:
1920
CUDA_VISIBLE_DEVICES: 0

0 commit comments

Comments
 (0)