24 lines
783 B
Docker
24 lines
783 B
Docker
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE=1
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
# 1. Системные зависимости
|
|
RUN apt-get update && apt-get install -y \
|
|
libglib2.0-0 libsm6 libxext6 libxrender-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# 2. Python пакеты
|
|
RUN pip install --no-cache-dir fastapi uvicorn timm scikit-learn pandas joblib python-multipart transformers==4.38.2 tokenizers==0.15.2 accelerate
|
|
|
|
# 3. Копируем код в контейнер
|
|
WORKDIR /app
|
|
COPY src/ /app/src/
|
|
|
|
# 4. МАГИЯ ЗДЕСЬ: Переходим внутрь папки src
|
|
WORKDIR /app/src
|
|
|
|
EXPOSE 8000
|
|
|
|
# 5. Запускаем локально (без префикса src.)
|
|
CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"] |