ci: improve docker for cu13 builds (#18194)

This commit is contained in:
ishandhanani
2026-02-03 13:09:38 -06:00
committed by GitHub
parent 0db6fd4dbe
commit 0a6925639b
2 changed files with 7 additions and 3 deletions

View File

@@ -19,13 +19,11 @@ ARG PIP_DEFAULT_INDEX
ARG UBUNTU_MIRROR
ARG GITHUB_ARTIFACTORY=github.com
ARG INSTALL_FLASHINFER_JIT_CACHE=0
ARG FLASHINFER_VERSION=0.6.1
ARG FLASHINFER_VERSION=0.6.2
ARG MOONCAKE_VERSION=0.3.8.post1
#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"
ARG FLASHINFER_VERSION=0.6.2
ENV DEBIAN_FRONTEND=noninteractive \
CUDA_HOME=/usr/local/cuda \
GDRCOPY_HOME=/usr/src/gdrdrv-${GDRCOPY_VERSION}/ \