Upgrade flashinfer-cubin and flashinfer-jit-cache versions

Signed-off-by: Zhou Taichang <tzhouam@connect.ust.hk>
This commit is contained in:
Zhou Taichang authored and GitHub committed 2026-03-16 19:31:03 +08:00
1 parent 8e81e60d32
commit 22a2bd3406
1 file changed
+2 -2
+2 -2
View File
@@ -21,12 +21,12 @@ RUN git clone https://github.com/vllm-project/vllm.git /workspace/vllm && \
cd ${APP_DIR} && \
uv pip install --system ".[dev]" && \
uv pip install --system --upgrade \
"flashinfer-cubin==0.6.4" \
"flashinfer-cubin==0.6.6" \
"nvidia-cublas-cu12==12.9.1.4" \
"numpy==2.2.6"
RUN uv pip install --system --upgrade \
"flashinfer-jit-cache==0.6.4" \
"flashinfer-jit-cache==0.6.6" \
--index-url https://flashinfer.ai/whl/cu129
RUN ln -sf /usr/bin/python3 /usr/bin/python