From 97fc950635392479593e955180a01fa14a8f14fa Mon Sep 17 00:00:00 2001 From: Teng Ma Date: Thu, 23 Apr 2026 04:36:10 +0800 Subject: [PATCH] [Docker] chore: update mooncake wheel version (#20731) Co-authored-by: Shangming Cai --- docker/Dockerfile | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 9892081b7..3ed313a01 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -20,7 +20,7 @@ ARG UBUNTU_MIRROR ARG GITHUB_ARTIFACTORY=github.com ARG INSTALL_FLASHINFER_JIT_CACHE=0 ARG FLASHINFER_VERSION=0.6.7.post3 -ARG MOONCAKE_VERSION=0.3.9 +ARG MOONCAKE_VERSION=0.3.10.post2 #if need other arg please add in MOONCAKE_COMPILE_ARG ARG MOONCAKE_COMPILE_ARG="-DUSE_HTTP=ON -DUSE_MNNVL=ON -DUSE_CUDA=ON -DWITH_EP=ON" @@ -502,17 +502,8 @@ RUN --mount=type=cache,target=/var/cache/apt,id=framework-apt \ RUN --mount=type=cache,target=/root/.cache/pip \ CUDA_MAJOR="${CUDA_VERSION%%.*}" && \ if [ "$CUDA_MAJOR" -ge 13 ]; then \ - echo "CUDA >= 13, installing mooncake-transfer-engine from source code"; \ - git clone --branch v${MOONCAKE_VERSION} --depth 1 https://github.com/kvcache-ai/Mooncake.git && \ - cd Mooncake && \ - bash dependencies.sh && \ - mkdir -p build && \ - cd build && \ - cmake .. ${MOONCAKE_COMPILE_ARG} && \ - make -j$(nproc) && \ - make install; \ + python3 -m pip install mooncake-transfer-engine-cuda13==${MOONCAKE_VERSION}; \ else \ - echo "CUDA < 13, installing mooncake-transfer-engine from pip"; \ python3 -m pip install mooncake-transfer-engine==${MOONCAKE_VERSION}; \ fi