diff --git a/Dockerfile b/Dockerfile index 1725213..8ab305b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,16 +2,11 @@ FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 ENV DEBIAN_FRONTEND=noninteractive \ PYTHONUNBUFFERED=1 \ - UV_SYSTEM_PYTHON=1 \ UV_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu121 \ PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True,max_split_size_mb:128 # Base OS tools + media stack + Python toolchain. RUN apt-get update && apt-get install -y --no-install-recommends \ - python3.10 \ - python3.10-dev \ - python3.10-venv \ - python3-pip \ ffmpeg \ git \ git-lfs \ @@ -28,6 +23,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ # Install uv. COPY --from=ghcr.io/astral-sh/uv:0.6.17 /uv /uvx /usr/local/bin/ +RUN uv python install 3.10 +# Place executables in the environment at the front of the path +ENV PATH="/app/.venv/bin:$PATH" WORKDIR /app