diff --git a/Dockerfile b/Dockerfile index a0d179d9a..03bbd3e3f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -118,7 +118,7 @@ RUN python3 -m pip install --no-cache-dir "cmake>=3.27,<4.0" scikit-build-core== export CUDAARCHS="80;86;89" && \ export CMAKE_CUDA_ARCHITECTURES="$CUDAARCHS" && \ # 这里保留常规参数;如果项目支持,也把内核编译线程设为 1(未知项将被忽略,不会报错) - export CMAKE_ARGS="-DCMAKE_CUDA_ARCHITECTURES=$CUDAARCHS -DSGL_KERNEL_COMPILE_THREADS=1 -Wno-dev" && \ + export CMAKE_ARGS="-DCMAKE_CUDA_ARCHITECTURES=$CUDAARCHS -DSGL_KERNEL_COMPILE_THREADS=8 -Wno-dev" && \ python3 -m pip wheel . --no-deps --no-build-isolation -w /tmp/sgl_kernel_wheels \ '