Skip to content

Commit a50e415

Browse files
committed
[refactor] dockerfile 수정
1 parent 0c69673 commit a50e415

File tree

2 files changed

+22
-26
lines changed

2 files changed

+22
-26
lines changed

Dockerfile

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,14 @@
11
# 1. 권장 베이스: CUDA 포함 PyTorch 런타임
22
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
33

4-
# OpenCV 헤드리스 등에 필요한 OS 패키지
4+
# OpenCV/Ultralytics가 필요로 하는 OS 라이브러리
55
RUN apt-get update && apt-get install -y --no-install-recommends \
6-
libglib2.0-0 libsm6 libxext6 libxrender1 && rm -rf /var/lib/apt/lists/*
6+
libgl1 \
7+
libglib2.0-0 \
8+
libsm6 \
9+
libxext6 \
10+
libxrender1 \
11+
&& rm -rf /var/lib/apt/lists/*
712

813
# HF 캐시 + PyTorch 메모리 튜닝(선택)
914
ENV HF_HOME=/models/hf-cache \
@@ -16,7 +21,7 @@ WORKDIR /app
1621
# 3. 의존성만 먼저 복사 → 캐시 최대 활용
1722
COPY everTale/requirements.txt ./requirements.txt
1823
RUN pip install --no-cache-dir --upgrade pip \
19-
&& pip install --no-cache-dir -r /app/requirements.txt opencv-python-headless ultralytics
24+
&& pip install --no-cache-dir -r /app/requirements.txt
2025

2126

2227
# 모델 파일을 이미지에 포함

everTale/requirements.txt

Lines changed: 14 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,19 @@
1-
# --- PyTorch stack (이미지에 포함되지만 버전 명시해두면 좋음) ---
2-
torch==2.3.1
3-
torchvision==0.18.1
4-
torchaudio==2.3.1
5-
6-
# --- Diffusers/Transformers/Accelerate (안정 조합) ---
1+
# --- Diffusers/Transformers/Accelerate ---
72
diffusers==0.29.0
83
transformers==4.42.4
94
accelerate==0.32.1
105
safetensors==0.4.3
116
huggingface-hub==0.23.4
127
tokenizers==0.19.1
8+
peft==0.11.1
139

14-
# --- API / 서버 ---
15-
fastapi==0.115.6 # (너가 쓰는 0.115.x 라인 유지, 너무 최신 patch는 피함)
16-
uvicorn==0.30.6 # 안정 버전
17-
starlette
10+
# --- API / Server ---
11+
fastapi==0.115.6
12+
uvicorn==0.30.6
13+
starlette==0.41.2 # ★ FastAPI 0.115.x와 호환 핀
1814

19-
# --- 유틸 ---
20-
numpy==1.26.4 # torch 2.3.x와 널리 쓰이는 안정 버전
15+
# --- Utils ---
16+
numpy==1.26.4
2117
pillow==10.4.0
2218
requests==2.32.3
2319
tqdm==4.66.4
@@ -28,18 +24,13 @@ PyYAML==6.0.2
2824
regex==2024.5.15
2925
typing_extensions==4.12.2
3026
fsspec==2024.6.1
31-
ultralytics
32-
33-
# --- 선택(LLM LoRA 등에 필요하면) ---
34-
peft==0.11.1
27+
pandas==2.3.1 # (정말 쓰면 유지, 아니면 삭제)
3528

36-
# --- 필요 시만 추가 (성공/실패 갈릴 수 있으므로 초기엔 제외 권장) ---
37-
# xformers==0.0.27.post2
29+
# --- YOLO / OpenCV ---
30+
ultralytics
31+
ultralytics-thop==2.0.15 # ★ YOLO가 내부 참조
32+
opencv-python-headless==4.12.0.88 # ★ 서버/컨테이너용 (GUI 의존 제거)
3833

39-
# --- OpenAI SDK (필수 추가) ---
34+
# --- Cloud / LLM ---
4035
openai>=1.30.0
41-
42-
# --- AWS SDK ---
4336
boto3==1.35.24
44-
45-

0 commit comments

Comments
 (0)