[Nemo CICD] Trigger on comment issued (#9062)

* match pytorch * match pytorch Signed-off-by: Ao Tang <[email protected]>
NVIDIA · May 2, 2024 · fb883ed · fb883ed
1 parent 6066774
commit fb883ed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -44,7 +44,7 @@ jobs:
 #  checkout-repository:
 #    runs-on: self-hosted-azure
 #    container:
-#      image: nvcr.io/nvidia/pytorch:24.01-py3
+#      image: nvcr.io/nvidia/pytorch:24.02-py3
 #      volumes:
 #        - ${{ github.workspace }}:/workspace
 #    steps:
@@ -60,7 +60,7 @@ jobs:
     if: ${{ github.event.label.name == 'Run CICD' }}
     # uses: actions/cache@v2
     #container:
-#      image: nvcr.io/nvidia/pytorch:24.01-py3
+#      image: nvcr.io/nvidia/pytorch:24.02-py3
 #      options: 
 #        # --user 0:128
 #        --device=/dev/nvidia0
@@ -78,7 +78,7 @@ jobs:
       run: |
         # Pull base PyTorch container
         docker pull nvcr.io/nvidia/pytorch:24.02-py3
-        docker run --device=/dev/nvidia0 --gpus all --shm-size=8g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume ${{ github.workspace }}/${{ github.run_id }}:/workspace --volume /mnt/datadrive/TestData:/home/TestData nvcr.io/nvidia/pytorch:24.01-py3 /bin/bash -c '
+        docker run --device=/dev/nvidia0 --gpus all --shm-size=8g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume ${{ github.workspace }}/${{ github.run_id }}:/workspace --volume /mnt/datadrive/TestData:/home/TestData nvcr.io/nvidia/pytorch:24.02-py3 /bin/bash -c '
             set -x
 
             # PyTorch version