FastAPI 기반 OCR 서비스 구현 및 Docker 환경 구성
This commit is contained in:
11
Dockerfile
11
Dockerfile
@@ -11,8 +11,6 @@ ENV DEBIAN_FRONTEND=noninteractive \
|
||||
TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas \
|
||||
TORCH_CUDA_ARCH_LIST="8.0"
|
||||
|
||||
WORKDIR /workspace
|
||||
|
||||
# 필수 빌드 도구 설치
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
git build-essential ninja-build \
|
||||
@@ -36,4 +34,11 @@ RUN pip install vllm==0.8.5
|
||||
RUN pip cache purge && \
|
||||
pip install --no-cache-dir --no-build-isolation --no-binary=flash-attn flash-attn==2.7.3
|
||||
|
||||
WORKDIR /workspace
|
||||
# API 서버 실행 포트 노출
|
||||
EXPOSE 11635
|
||||
|
||||
WORKDIR /workspace
|
||||
COPY . .
|
||||
|
||||
# Uvicorn으로 FastAPI 서버 실행
|
||||
CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "11635"]
|
||||
Reference in New Issue
Block a user