Add -y flag to pip uninstall pynvml for non-interactive Docker build

2026-04-03 10:57:42 +00:00
parent 5280a28205
commit 2442906d95
1 changed files with 1 additions and 1 deletions
--- 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"]