Upgrade flashinfer-cubin and flashinfer-jit-cache versions
Signed-off-by: Zhou Taichang <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
8e81e60d32
commit
22a2bd3406
1 file changed
+2
-2
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user