numpy version fix

2025-04-03 20:35:56 +00:00
parent 75e33490bd
commit 4d0dc5d06f
1 changed files with 1 additions and 0 deletions
--- a/vllm/Dockerfile
+++ b/vllm/Dockerfile
@@ -45,6 +45,7 @@ ENV PATH=${VIRTUAL_ENV}/bin:${PATH}
 ENV CUDA_HOME=/usr/local/cuda
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}

+RUN uv pip install numpy==2.0.0
 # Install pytorch nightly
 RUN uv pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126