diff --git a/.github/workflows/gpu.yml b/.github/workflows/gpu.yml index a763013af3..5ecce1a839 100644 --- 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: