From ebe927fadcbdf9d02bc71f55d64fa5f23850425f Mon Sep 17 00:00:00 2001 From: Jeremy Dyer Date: Tue, 5 Nov 2024 16:09:04 -0500 Subject: [PATCH] Add test file to test gpu runners --- .github/workflows/test-gpu-runner.yml | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 .github/workflows/test-gpu-runner.yml diff --git a/.github/workflows/test-gpu-runner.yml b/.github/workflows/test-gpu-runner.yml new file mode 100644 index 00000000..fc6ec53c --- /dev/null +++ b/.github/workflows/test-gpu-runner.yml @@ -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