o2o-plagiarism-ai/Dockerfile.gpu-embed

9 lines
298 B
Docker

# 운영 API 이미지와 분리된 2차 KoSimCSE 배치 전용 이미지.
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt sentence-transformers==3.0.1
COPY app ./app
COPY scripts ./scripts
COPY data ./data