From 178b178ef54eb58ca0df92ee368a2b0eca62daa1 Mon Sep 17 00:00:00 2001 From: Oliver Koenig Date: Fri, 25 Oct 2024 10:07:13 +0200 Subject: [PATCH] f Signed-off-by: Oliver Koenig --- .github/workflows/_test_template.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/_test_template.yml b/.github/workflows/_test_template.yml index 047f035f8bc2f..56623c8fbff04 100644 --- a/.github/workflows/_test_template.yml +++ b/.github/workflows/_test_template.yml @@ -56,12 +56,12 @@ jobs: - name: Start container run: | if [[ "${{ inputs.RUNNER }}" == *cpu* ]]; then - ARG=("--gpus all") + ARG=("--runtime=nvidia --gpus all") else ARG=("") fi - docker run --rm -d --name nemo_container_${{ github.run_id }} --runtime=nvidia ${ARG[@]} --shm-size=64g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume /mnt/datadrive/TestData:/home/TestData nemoci.azurecr.io/nemo_container_${{ github.run_id }} bash -c "sleep $(( ${{ inputs.TIMEOUT }} * 60 + 60 ))" + docker run --rm -d --name nemo_container_${{ github.run_id }} ${ARG[@]} --shm-size=64g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume /mnt/datadrive/TestData:/home/TestData nemoci.azurecr.io/nemo_container_${{ github.run_id }} bash -c "sleep $(( ${{ inputs.TIMEOUT }} * 60 + 60 ))" - id: main name: Run main script