diff --git a/Containerfile b/Containerfile index 46b773a..354300a 100644 --- a/Containerfile +++ b/Containerfile @@ -29,7 +29,7 @@ ARG TORCH_INDEX=https://download.pytorch.org/whl/cu128 RUN pip install torch==${TORCH_VERSION} --extra-index-url ${TORCH_INDEX} ARG FLASH_ATTN_VERSION=2.8.3 -ARG MAX_JOBS=4 +ARG MAX_JOBS=2 RUN mkdir /wheels && \ MAX_JOBS=${MAX_JOBS} pip wheel flash-attn==${FLASH_ATTN_VERSION} \ diff --git a/push-to-gitea.sh b/push-to-gitea.sh index 320d918..43c1382 100755 --- a/push-to-gitea.sh +++ b/push-to-gitea.sh @@ -46,7 +46,10 @@ case "$1" in --build|*) echo "==> flash-attn-builder 이미지 빌드..." cd "$(dirname "$0")" + # --cpu-shares 512: 다른 서비스 대비 절반 우선순위 (유휴 시 자동 해제) + # --build-arg MAX_JOBS=2: 컴파일러 병렬도 제한 (기본 2, 조정 가능) podman build \ + --cpu-shares 512 \ --tag "localhost/flash-attn-builder:${TAG}" \ --file Containerfile \ .