fix: MAX_JOBS 4→2, cpu-shares 512로 서비스 영향 최소화
This commit is contained in:
@@ -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} \
|
||||
|
||||
@@ -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 \
|
||||
.
|
||||
|
||||
Reference in New Issue
Block a user