Update gpu-ci.yml to add container (NVIDIA-Merlin#1066)

CosmicMedia · Sep 19, 2023 · aba6aa2 · aba6aa2
1 parent 844c7f3
commit aba6aa2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
@@ -12,6 +12,8 @@ on:
 jobs:
   gpu-ci:
     runs-on: linux-amd64-gpu-p100-latest-1
+    container:
+          image: nvcr.io/nvidia/merlin/merlin-tensorflow:nightly
     steps:
     - uses: actions/checkout@v3
       with: