diff --git a/backend-gpu/base/Dockerfile b/backend-gpu/base/Dockerfile
index a2c98cf..23cdafd 100644
--- a/backend-gpu/base/Dockerfile
+++ b/backend-gpu/base/Dockerfile
@@ -98,7 +98,7 @@ RUN git clone --depth 1 --branch 'master' https://github.com/davisking/dlib.git
 RUN ls -l /usr/local/lib/python3.12/dist-packages
 
 # Stage 2: Deployment Stage
-FROM ubuntu:noble
+FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04
 ARG TARGETPLATFORM
 ENV DEBIAN_FRONTEND=noninteractive
 WORKDIR /app