v0.10.2 cleanup

2025-09-24 06:14:16 +00:00
parent fc321295f1
commit 201bbf5379
1 changed files with 3 additions and 0 deletions
--- a/vllm/Dockerfile
+++ b/vllm/Dockerfile
@@ -174,6 +174,9 @@ RUN apt-get update && apt install -y wget libglib2.0-0
 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 uv pip uninstall pynvml && uv pip install nvidia-ml-py
+
 # API server entrypoint
 # ENTRYPOINT ["vllm", "serve"]
 CMD ["/bin/bash"]