diff --git a/.github/workflows/gpu_benchmark.yml b/.github/workflows/gpu_benchmark.yml
index d3480c112..7fc1dd23f 100644
--- a/.github/workflows/gpu_benchmark.yml
+++ b/.github/workflows/gpu_benchmark.yml
@@ -15,7 +15,7 @@ jobs:
     runs-on: self-hosted
     container:
       image: ghcr.io/nvidia/jax:jax
-      options: --rm --gpus all
+      options: --rm --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864
 
     steps:
       - name: Checkout repository