Update gpu-ci.yml add nvidia env var

NVIDIA-Merlin · Sep 20, 2023 · 103b058 · 103b058
1 parent aba6aa2
commit 103b058
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
@@ -14,6 +14,8 @@ jobs:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
           image: nvcr.io/nvidia/merlin/merlin-tensorflow:nightly
+          env:
+            NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     steps:
     - uses: actions/checkout@v3
       with: