diff --git a/Dockerfile b/Dockerfile index ae292d7f9..da26ab0fa 100644 --- a/Dockerfile +++ b/Dockerfile @@ -99,7 +99,7 @@ RUN python3 -m pip install ".[srt,openai]" --no-build-isolation && \ # ── 用你本地源码编 sgl-kernel==0.3.9.post2(与自编 torch 完全 ABI 对齐) ────── WORKDIR /sgl/sglang/sgl-kernel -RUN python3 -m pip install --no-cache-dir scikit-build-core pybind11[global] packaging && \ +RUN python3 -m pip install --no-cache-dir "cmake>=3.27" scikit-build-core pybind11[global] packaging && \ bash -lc 'export CMAKE_PREFIX_PATH="$(python3 -c "import torch; print(torch.utils.cmake_prefix_path)")" \ && export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9;9.0" SGL_KERNEL_COMPILE_THREADS=1 CMAKE_BUILD_PARALLEL_LEVEL=${MAX_JOBS:-96} FORCE_CUDA=1 \ && python3 -m pip wheel . --no-deps --no-build-isolation -w /tmp/sgl_kernel_wheels'