mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-28 10:07:03 +08:00
Update docker/Dockerfile
Signed-off-by: Pavani Majety <pavanimajety@gmail.com>
This commit is contained in:
parent
20d87ee2f0
commit
4f1aa77ff9
@ -466,7 +466,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
|
|||||||
# Install FlashInfer pre-compiled kernel cache and binaries
|
# Install FlashInfer pre-compiled kernel cache and binaries
|
||||||
# This is ~1.1GB and only changes when FlashInfer version bumps
|
# This is ~1.1GB and only changes when FlashInfer version bumps
|
||||||
# https://docs.flashinfer.ai/installation.html
|
# https://docs.flashinfer.ai/installation.html
|
||||||
ARG FLASHINFER_VERSION=0.6.0rc1
|
ARG FLASHINFER_VERSION=0.6.0rc2
|
||||||
RUN --mount=type=cache,target=/root/.cache/uv \
|
RUN --mount=type=cache,target=/root/.cache/uv \
|
||||||
uv pip install --system flashinfer-cubin==${FLASHINFER_VERSION} \
|
uv pip install --system flashinfer-cubin==${FLASHINFER_VERSION} \
|
||||||
--extra-index-url https://flashinfer.ai/whl \
|
--extra-index-url https://flashinfer.ai/whl \
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user