bvandeusen
7b10f4caab
onnxruntime-gpu needs cuDNN 9; the plain cuda:12.4.1-runtime image lacks it (libcudnn.so.9 missing → CUDAExecutionProvider falls back to CPU). Switch to the -cudnn-runtime variant which bundles cuDNN 9. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Ttrj5P7upUTueSfoJcxEqa
23 lines
868 B
Docker
23 lines
868 B
Docker
# FabledCurator GPU agent — runs on the desktop with the GPU.
|
|
# CUDA + cuDNN runtime so onnxruntime-gpu can use the card (it needs cuDNN 9 —
|
|
# the plain -runtime image lacks it: "libcudnn.so.9: cannot open shared object
|
|
# file"); ffmpeg for video frames.
|
|
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1
|
|
RUN apt-get update \
|
|
&& apt-get install -y --no-install-recommends python3 python3-pip ffmpeg \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
COPY requirements.txt .
|
|
RUN pip3 install --no-cache-dir -r requirements.txt
|
|
COPY fc_agent ./fc_agent
|
|
|
|
# imgutils caches downloaded ONNX models here; mount a volume to persist them.
|
|
ENV HF_HOME=/models
|
|
EXPOSE 8770
|
|
|
|
# The control UI; the worker is started from it (or POST /start).
|
|
CMD ["uvicorn", "fc_agent.app:app", "--host", "0.0.0.0", "--port", "8770"]
|