20 lines
580 B
Docker
20 lines
580 B
Docker
# torch pip wheels bundle their own CUDA runtime libs; only the host NVIDIA
|
|
# driver is needed (exposed via the nvidia container runtime), so a plain
|
|
# Python base is sufficient — no CUDA base image required.
|
|
FROM python:3.11-slim
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends ffmpeg \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
COPY pyproject.toml ./
|
|
COPY src ./src
|
|
RUN pip install --no-cache-dir .
|
|
|
|
ENV SONGTRACE_CONFIG=/config/config.toml \
|
|
TORCH_HOME=/data/models \
|
|
XDG_CACHE_HOME=/data/models/cache
|
|
|
|
EXPOSE 8000
|
|
CMD ["songtrace", "serve"]
|