Test CI

Signed-off-by: Vibhu Jawa <[email protected]>
rapidsai · Jul 17, 2024 · 51b0077 · 51b0077
1 parent 356b708
commit 51b0077
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
@@ -13,10 +13,10 @@ jobs:
   gpu-ci:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidian/crossfit-ci:23.10
+      image: rapidsai/base:24.06-cuda12.2-py3.11
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
-      options: --shm-size=1G
+      options: --shm-size=1G -ulimit memlock=-1 --ulimit stack=67108864
 
     steps:
       - uses: actions/checkout@v3