From b036c487e469eb04da26ac0882733719b7f761a3 Mon Sep 17 00:00:00 2001 From: hailin Date: Thu, 18 Sep 2025 19:42:46 +0800 Subject: [PATCH] . --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 59be356a0..0684a4aab 100644 --- a/Dockerfile +++ b/Dockerfile @@ -101,7 +101,7 @@ RUN python3 -m pip install ".[srt,openai]" --no-build-isolation && \ WORKDIR /sgl/sglang/sgl-kernel RUN python3 -m pip install --no-cache-dir "cmake>=3.27,<4.0" scikit-build-core==0.11.6 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 \ + && export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9" SGL_KERNEL_COMPILE_THREADS=1 CMAKE_BUILD_PARALLEL_LEVEL=${MAX_JOBS:-96} FORCE_CUDA=1 \ && export CMAKE_ARGS="-DCMAKE_POLICY_VERSION_MINIMUM=3.5" \ && python3 -m pip wheel . --no-deps --no-build-isolation -w /tmp/sgl_kernel_wheels'