Uninstall old nvidia-cublas-cu12 version before reinstalling the pinned version to ensure consistent dependency management in Dockerfile.ci.
Signed-off-by: Taichang Zhou <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
169d94dab9
commit
514bd46f2e
1 file changed
+2
-1
@@ -30,7 +30,8 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm
|
||||
# "flashinfer-jit-cache==${FLASHINFER_VERSION}" \
|
||||
# --extra-index-url "https://flashinfer.ai/whl/cu${FLASHINFER_CUDA_TAG}" && \
|
||||
# flashinfer show-config
|
||||
|
||||
# Uninstall the old nvidia-cublas-cu12 brought by torch/vllm before reinstalling the pinned version
|
||||
RUN uv pip uninstall --system nvidia-cublas-cu12
|
||||
# Install vllm-omni into the same uv-managed Python environment used by the base image.
|
||||
# Use bash -c so that $(python3 -c ...) is expanded inside the container.
|
||||
RUN /bin/bash -c 'uv pip install --python "$(python3 -c "import sys; print(sys.executable)")" --no-cache-dir -e .'
|
||||
|
||||
Reference in new issue
Block a user