Update Dockerfile.ci to install additional dependencies including flashinfer-jit-cache, nvidia-cublas-cu12, and numpy, ensuring a more comprehensive environment setup for development.
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
98926707f5
commit
8dcf1061f1
1 file changed
+5
-2
@@ -21,8 +21,11 @@ RUN apt-get update && \
|
||||
RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/6521ccf2860eb72cf23a9fa9044bf9721fd9a7c6
|
||||
|
||||
RUN uv pip install --system --no-cache-dir ".[dev]"
|
||||
RUN uv pip install --system --upgrade "flashinfer-cubin==0.6.4"
|
||||
|
||||
RUN uv pip install --system --upgrade \
|
||||
"flashinfer-cubin==0.6.4" \
|
||||
"flashinfer-jit-cache==0.6.4" \
|
||||
"nvidia-cublas-cu12==12.9.1.4" \
|
||||
"numpy==2.2.6"
|
||||
RUN ln -sf /usr/bin/python3 /usr/bin/python
|
||||
|
||||
ENTRYPOINT []
|
||||
Reference in new issue
Block a user