Refactor Dockerfile.ci to streamline package installations by using 'uv pip' consistently for both 'pplx_kernels' and VLLM dependencies, ensuring a more uniform CI environment setup.

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-02-13 06:55:10 +00:00
1 parent 3abec0173c
commit 963deaf5ad
1 file changed
+2 -2
+2 -2
View File
@@ -24,9 +24,9 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm
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 uv pip install --python "$(python3 -c 'import sys; print(sys.executable)')" --no-cache-dir ".[dev]"
RUN uv pip install --system --no-cache-dir ".[dev]"
RUN pip install --system --upgrade --force-reinstall "pplx_kernels"
RUN uv pip install --system --upgrade --force-reinstall "pplx_kernels"
RUN /bin/bash -c 'FLASHINFER_CUDA_TAG="$(python3 -c "import torch; print((torch.version.cuda or \"12.4\").replace(\".\", \"\"))")" && \
uv pip install --system --upgrade --force-reinstall \