diff --git a/ci/base.yml b/ci/base.yml
index 6d9cc48c74..fb2987e763 100644
--- a/ci/base.yml
+++ b/ci/base.yml
@@ -61,7 +61,8 @@ build_image_aarch64:
   variables:
     SLURM_JOB_NUM_NODES: 1
     SLURM_NTASKS: 1
-    SLURM_GPUS_PER_NODE: 4
+    SLURM_GPUS_PER_NODE: 1
+    SLURM_PARTITION: "shared"
     SLURM_TIMELIMIT: '01:30:00'
     CSCS_CUDA_MPS: 1
     NUM_PROCESSES: auto