diff --git a/Dockerfile.ubi b/Dockerfile.ubi index 50f2b5d114740..cff1f9351c698 100644 --- a/Dockerfile.ubi +++ b/Dockerfile.ubi @@ -174,7 +174,7 @@ RUN --mount=type=bind,from=libsodium-builder,src=/usr/src/libsodium,target=/usr/ RUN --mount=type=cache,target=/root/.cache/pip \ --mount=type=cache,target=/root/.cache/uv \ uv pip install \ - "https://github.com/flashinfer-ai/flashinfer/releases/download/v0.1.6/flashinfer-0.1.6+cu124torch2.4-cp312-cp312-linux_x86_64.whl" + "https://github.com/flashinfer-ai/flashinfer/releases/download/v0.2.0.post2/flashinfer_python-0.2.0.post2+cu124torch2.5-cp312-cp312-linux_x86_64.whl" ENV HF_HUB_OFFLINE=1 \ HOME=/home/vllm \