Skip to content

Commit

Permalink
Add test file to test gpu runners
Browse files Browse the repository at this point in the history
  • Loading branch information
jdye64 committed Nov 5, 2024
1 parent fa07a5c commit ebe927f
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions .github/workflows/test-gpu-runner.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
name: Test Self Hosted Runners
on: push
jobs:
job1_cpu:
runs-on: linux-amd64-cpu8
steps:
- name: hello
run: echo "hello"
job2_gpu:
runs-on: linux-amd64-gpu-a100-latest-1-testing
container: # GPU jobs must run in a container
image: nvidia/cuda:12.0.0-base-ubuntu22.04
env:
NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }} # GPU jobs must set this container env variable
steps:
- name: hello
run: |
echo "hello"
nvidia-smi

0 comments on commit ebe927f

Please sign in to comment.