diff --git a/vllm/Dockerfile b/vllm/Dockerfile
index 79ad8ee..cef70f0 100644
--- a/vllm/Dockerfile
+++ b/vllm/Dockerfile
@@ -198,7 +198,7 @@ RUN wget ${NSYS_URL}${NSYS_PKG} && dpkg -i $NSYS_PKG && rm $NSYS_PKG
 RUN apt install -y --no-install-recommends tmux cmake
 
 # Deprecated cleanup
-RUN pip uninstall pynvml && pip install nvidia-ml-py
+RUN pip uninstall -y pynvml && pip install nvidia-ml-py
 
 # API server entrypoint
 # ENTRYPOINT ["vllm", "serve"]