Try to set NVIDIA_VISIBLE_DEVICES

Signed-off-by: Ben Howe <[email protected]>
NVIDIA · Jan 8, 2025 · 4bbe5e6 · 4bbe5e6
1 parent d1a6276
commit 4bbe5e6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/.github/workflows/all_libs.yaml b/.github/workflows/all_libs.yaml
@@ -9,7 +9,10 @@ jobs:
     if: startsWith(github.ref, 'refs/heads/pull-request/')
     runs-on: linux-arm64-gpu-a100-latest-1
     #runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-arm64-cpu8' || 'ubuntu-latest' }}
-    container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
+    container:
+      image: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
+      env:
+        NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     permissions:
       actions: write
       contents: read

diff --git a/.github/workflows/lib_qec.yaml b/.github/workflows/lib_qec.yaml
@@ -9,7 +9,10 @@ jobs:
     if: startsWith(github.ref, 'refs/heads/pull-request/')
     runs-on: linux-arm64-gpu-a100-latest-1
     #runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-arm64-cpu8' || 'ubuntu-latest' }}
-    container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
+    container:
+      image: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
+      env:
+        NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     permissions:
       actions: write
       contents: read

diff --git a/.github/workflows/lib_solvers.yaml b/.github/workflows/lib_solvers.yaml
@@ -9,7 +9,10 @@ jobs:
     if: startsWith(github.ref, 'refs/heads/pull-request/')
     runs-on: linux-arm64-gpu-a100-latest-1
     #runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-arm64-cpu8' || 'ubuntu-latest' }}
-    container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
+    container:
+      image: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
+      env:
+        NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     permissions:
       actions: write
       contents: read