diff --git a/presto-native-execution/scripts/dockerfiles/centos-dependency.dockerfile b/presto-native-execution/scripts/dockerfiles/centos-dependency.dockerfile
index fc9c0c810f265..bf2c2f50398a2 100644
--- a/presto-native-execution/scripts/dockerfiles/centos-dependency.dockerfile
+++ b/presto-native-execution/scripts/dockerfiles/centos-dependency.dockerfile
@@ -16,10 +16,14 @@ FROM quay.io/centos/centos:stream9
 # from https://github.com/facebookincubator/velox/pull/14366
 ARG ARM_BUILD_TARGET
 
+# This defaults to 12.9 but can be overridden with a build arg
+ARG CUDA_VERSION
+
 ENV PROMPT_ALWAYS_RESPOND=y
 ENV CC=/opt/rh/gcc-toolset-12/root/bin/gcc
 ENV CXX=/opt/rh/gcc-toolset-12/root/bin/g++
 ENV ARM_BUILD_TARGET=${ARM_BUILD_TARGET}
+ENV CUDA_VERSION=${CUDA_VERSION:-12.9}
 
 RUN mkdir -p /scripts /velox/scripts
 COPY scripts /scripts
@@ -35,7 +39,7 @@ RUN bash -c "mkdir build && \
                  source ../velox/scripts/setup-centos-adapters.sh && \
                  install_adapters && \
                  install_clang15 && \
-                 install_cuda 12.8) && \
+                 install_cuda ${CUDA_VERSION}) && \
     rm -rf build"
 
 # put CUDA binaries on the PATH