File tree Expand file tree Collapse file tree 2 files changed +22
-26
lines changed
Expand file tree Collapse file tree 2 files changed +22
-26
lines changed Original file line number Diff line number Diff line change 11# 1. 권장 베이스: CUDA 포함 PyTorch 런타임
22FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
33
4- # OpenCV 헤드리스 등에 필요한 OS 패키지
4+ # OpenCV/Ultralytics가 필요로 하는 OS 라이브러리
55RUN 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 메모리 튜닝(선택)
914ENV HF_HOME=/models/hf-cache \
@@ -16,7 +21,7 @@ WORKDIR /app
1621# 3. 의존성만 먼저 복사 → 캐시 최대 활용
1722COPY everTale/requirements.txt ./requirements.txt
1823RUN 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# 모델 파일을 이미지에 포함
Original file line number Diff line number Diff line change 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 ---
72diffusers == 0.29.0
83transformers == 4.42.4
94accelerate == 0.32.1
105safetensors == 0.4.3
116huggingface-hub == 0.23.4
127tokenizers == 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
2117pillow == 10.4.0
2218requests == 2.32.3
2319tqdm == 4.66.4
@@ -28,18 +24,13 @@ PyYAML==6.0.2
2824regex == 2024.5.15
2925typing_extensions == 4.12.2
3026fsspec == 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 ---
4035openai >= 1.30.0
41-
42- # --- AWS SDK ---
4336boto3 == 1.35.24
44-
45-
You can’t perform that action at this time.
0 commit comments