25 lines
827 B
Docker
25 lines
827 B
Docker
FROM python:3.12-slim
|
|
|
|
# системные зависимости для аудио-декодинга
|
|
RUN apt-get update && \
|
|
apt-get install -y --no-install-recommends ffmpeg curl && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
# NVIDIA runtime libraries (cuBLAS + cuDNN) — ставим через pip,
|
|
# чтобы не тянуть полный CUDA Toolkit в образ
|
|
RUN pip install --no-cache-dir nvidia-cublas-cu12 "nvidia-cudnn-cu12==9.*"
|
|
|
|
# LD_LIBRARY_PATH для CTranslate2
|
|
ENV LD_LIBRARY_PATH="/usr/local/lib/python3.12/site-packages/nvidia/cublas/lib:/usr/local/lib/python3.12/site-packages/nvidia/cudnn/lib:${LD_LIBRARY_PATH}"
|
|
|
|
COPY app.py .
|
|
|
|
EXPOSE 8001
|
|
|
|
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]
|