diff --git a/Dockerfile b/Dockerfile index 8b352a1..5f11fa7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -4,11 +4,15 @@ FROM registry.fedoraproject.org/fedora:43 ARG HTTP_PROXY="" ARG HTTPS_PROXY="" ARG SOCKS_PROXY="" +ARG MAX_JOBS="32" # Set environment variables if proxies are provided ENV http_proxy=$HTTP_PROXY ENV https_proxy=$HTTPS_PROXY ENV no_proxy="localhost,127.0.0.1,::1" +ENV MAX_JOBS=$MAX_JOBS +ENV CMAKE_BUILD_PARALLEL_LEVEL=$MAX_JOBS +ENV NINJAFLAGS="-j$MAX_JOBS" # Configure DNF proxy if provided RUN if [ -n "$HTTP_PROXY" ]; then \ @@ -167,7 +171,7 @@ ENV VLLM_TARGET_DEVICE="rocm" ENV PYTORCH_ROCM_ARCH="gfx1201" ENV HIP_ARCHITECTURES="gfx1201" ENV AMDGPU_TARGETS="gfx1201" -ENV MAX_JOBS="32" +ENV MAX_JOBS=$MAX_JOBS ENV CC="/opt/rocm/llvm/bin/clang" ENV CXX="/opt/rocm/llvm/bin/clang++" diff --git a/build_and_run.sh b/build_and_run.sh index 29e0aa1..ea73fa3 100644 --- a/build_and_run.sh +++ b/build_and_run.sh @@ -330,7 +330,7 @@ build_image() { BUILD_START=$(date +%s) # 准备构建参数 - BUILD_ARGS="--network=host --build-arg MAX_JOBS=32" + BUILD_ARGS="--network=host --build-arg MAX_JOBS=${MAX_JOBS:-32}" if [ -n "$HTTP_PROXY_ARG" ]; then BUILD_ARGS="${BUILD_ARGS} --build-arg HTTP_PROXY=${HTTP_PROXY_ARG} --build-arg HTTPS_PROXY=${HTTP_PROXY_ARG}" fi