🚧 cuda-runtime?

Signed-off-by: Prashant Gupta <prashantgupta@us.ibm.com>
IBM · Apr 17, 2024 · 4686e0d · 4686e0d
1 parent e784d65
commit 4686e0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.ubi b/Dockerfile.ubi
@@ -261,7 +261,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 ## vLLM installation IMAGE ##########################################################
 # image with vLLM installed
-FROM cuda-base AS vllm
+FROM cuda-runtime AS vllm
 
 WORKDIR /vllm-workspace