using private container test

NVIDIA-Merlin · Oct 14, 2023 · e10d541 · e10d541
1 parent 49795c2
commit e10d541
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/gpu.yml b/.github/workflows/gpu.yml
@@ -13,10 +13,14 @@ jobs:
   gpu-ci-tensorflow:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidia/merlin/merlin-tensorflow:nightly
+      image: nvcr.io/nvidia/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: