diff --git a/Dockerfile-cuda b/Dockerfile-cuda
index a785e5f0..5bd0c377 100644
--- a/Dockerfile-cuda
+++ b/Dockerfile-cuda
@@ -1,4 +1,5 @@
 # syntax=docker/dockerfile:1
+ARG CUDA_VERSION=12.3.2
 
 ######## WebUI frontend ########
 FROM node:21-alpine3.19 as build
@@ -23,8 +24,8 @@ RUN npm run build
 #RUN OPENWEBUI_CPU_TARGET="cpu_avx2" sh gen_linux.sh
 
 ######## CUDA WebUI backend ########
-ARG CUDA_VERSION=12.3.2
-FROM --platform=linux/amd64 nvidia/cuda:$CUDA_VERSION-devel-ubuntu22.04 AS cuda-build-amd64
+FROM --platform=linux/amd64 nvidia/cuda:"$CUDA_VERSION"-devel-ubuntu22.04 AS cuda-build-amd64
+
 # Set environment variables for NVIDIA Container Toolkit
 ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 \
     NVIDIA_DRIVER_CAPABILITIES=all \