Skip to content

Commit 66faaec

Browse files
Update GitHub Actions workflow for GPU testing
1 parent 0b88c06 commit 66faaec

File tree

2 files changed

+30
-40
lines changed

2 files changed

+30
-40
lines changed
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
name: Test - P4D 4 GPUs
2+
3+
on:
4+
workflow_dispatch:
5+
push:
6+
paths:
7+
- ".github/workflows/test-cdk-changes.yaml"
8+
9+
jobs:
10+
test-gpu-standard:
11+
runs-on: gpu-standard-runners
12+
steps:
13+
- name: Check GPUs
14+
run: nvidia-smi
15+
16+
- name: Check Docker
17+
run: |
18+
docker info
19+
docker run --rm --gpus all nvidia/cuda:12.0-base nvidia-smi
20+
21+
test-p4d:
22+
runs-on: p4d-gpu-runners
23+
steps:
24+
- name: Check GPUs
25+
run: nvidia-smi
26+
27+
- name: Check Docker
28+
run: |
29+
docker info
30+
docker run --rm --gpus all nvidia/cuda:12.0-base nvidia-smi

.github/workflows/test-p4d-4gpus.yaml

Lines changed: 0 additions & 40 deletions
This file was deleted.

0 commit comments

Comments
 (0)