diff --git a/Dockerfile b/Dockerfile
index 44bc86c..2665061 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 FROM ghcr.io/astral-sh/uv:latest AS uv_bin
-FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
+FROM nvidia/cuda:12.4.1-base-ubuntu22.04
 
 ENV UV_LINK_MODE=copy \
     UV_COMPILE_BYTECODE=1 \