hailin
/
sglang_v0.5.2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-09-18 19:42:46 +08:00
parent 0992c81029
commit b036c487e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -101,7 +101,7 @@ RUN python3 -m pip install ".[srt,openai]" --no-build-isolation && \
WORKDIR /sgl/sglang/sgl-kernel 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 && \ 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)")" \ 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" \ && export CMAKE_ARGS="-DCMAKE_POLICY_VERSION_MINIMUM=3.5" \
&& python3 -m pip wheel . --no-deps --no-build-isolation -w /tmp/sgl_kernel_wheels' && python3 -m pip wheel . --no-deps --no-build-isolation -w /tmp/sgl_kernel_wheels'