f

Signed-off-by: Oliver Koenig <[email protected]>
NVIDIA · Oct 25, 2024 · 178b178 · 178b178
1 parent 054cb40
commit 178b178
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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