Use devel image for nvvm

NVIDIA-Merlin · Jun 2, 2023 · fef5b4e · fef5b4e
1 parent 475b96b
commit fef5b4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/gpu-tests.yml b/.github/workflows/gpu-tests.yml
@@ -17,7 +17,7 @@ jobs:
   gpu-tests:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvidia/cuda:11.8.0-base-ubuntu22.04
+      image: nvidia/cuda:11.8.0-devel-ubuntu22.04
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     steps: