add new container pull logic to all test sets

NVIDIA-Merlin · Oct 14, 2023 · c4c0ef0 · c4c0ef0
1 parent f50eb4e
commit c4c0ef0
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/.github/workflows/gpu.yml b/.github/workflows/gpu.yml
@@ -44,10 +44,13 @@ jobs:
   gpu-ci-pytorch:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidia/merlin/merlin-pytorch:nightly
+      image: nvcr.io/nvstaging/merlin/merlin-ci-runner:latest
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G
+      credentials:
+        username: $oauthtoken
+        password: ${{ secrets.NGC_TOKEN }}
     steps:
       - uses: actions/checkout@v3
         with:
@@ -71,10 +74,13 @@ jobs:
   gpu-ci-tensorflow-examples:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidia/merlin/merlin-tensorflow:nightly
+      image: nvcr.io/nvstaging/merlin/merlin-ci-runner:latest
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G
+      credentials:
+        username: $oauthtoken
+        password: ${{ secrets.NGC_TOKEN }}
     steps:
       - uses: actions/checkout@v3
         with:
@@ -97,10 +103,13 @@ jobs:
   gpu-ci-pytorch-examples:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidia/merlin/merlin-pytorch:nightly
+      image: nvcr.io/nvstaging/merlin/merlin-ci-runner:latest
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G
+      credentials:
+        username: $oauthtoken
+        password: ${{ secrets.NGC_TOKEN }}
     steps:
       - uses: actions/checkout@v3
         with: