diff --git a/images/api-base/Dockerfile b/images/api-base/Dockerfile
index 15d7211..5845599 100644
--- a/images/api-base/Dockerfile
+++ b/images/api-base/Dockerfile
@@ -1,7 +1,4 @@
-
-# FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
-
-FROM nvidia/cuda:12.2.2-devel-ubuntu22.04
+FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
 RUN apt-get update && \
     apt-get install -y python3-pip python3-dev && \
     rm -rf /var/lib/apt/lists/*