Update Dockerfile.ci to install additional dependencies for flashinfer and numpy, and create a symlink for python3. This enhances the CI environment setup.

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-03-10 08:05:28 +00:00
1 parent 8ac369c84a
commit cf0a7a5475
1 file changed
+9
+9
View File
@@ -21,4 +21,13 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm
RUN uv pip install --system --no-cache-dir ".[dev]"
RUN uv pip install --system --upgrade \
"flashinfer-cubin==0.6.4" \
"nvidia-cublas-cu12==12.9.1.4" \
"numpy==2.2.6"
RUN uv pip install --system --upgrade \
"flashinfer-jit-cache==0.6.4" \
--index-url https://flashinfer.ai/whl/cu129
RUN ln -sf /usr/bin/python3 /usr/bin/python
ENTRYPOINT []