Dockerfile.ubi: change release stage base to python-cuda-base

dtrifiro · dtrifiro · commit ad5028efdbec · 2024-08-13T01:42:33.000+02:00
diff --git a/Dockerfile.ubi b/Dockerfile.ubi
@@ -153,7 +153,7 @@ RUN curl -LO https://github.com/jedisct1/libsodium/releases/download/${LIBSODIUM
 RUN ./configure --prefix="/usr/" && make && make check
 
 ## Release #####################################################################
-FROM python-install AS vllm-openai
+FROM python-cuda-base AS vllm-openai
 
 WORKDIR /workspace
 
