diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
index 36662ec..db95905 100644
--- 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