forked from LiveCarta/ContentGeneration
Updated uv paths, do not use system python
This commit is contained in:
@@ -2,16 +2,11 @@ FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
|
|||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive \
|
ENV DEBIAN_FRONTEND=noninteractive \
|
||||||
PYTHONUNBUFFERED=1 \
|
PYTHONUNBUFFERED=1 \
|
||||||
UV_SYSTEM_PYTHON=1 \
|
|
||||||
UV_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu121 \
|
UV_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu121 \
|
||||||
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True,max_split_size_mb:128
|
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True,max_split_size_mb:128
|
||||||
|
|
||||||
# Base OS tools + media stack + Python toolchain.
|
# Base OS tools + media stack + Python toolchain.
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
python3.10 \
|
|
||||||
python3.10-dev \
|
|
||||||
python3.10-venv \
|
|
||||||
python3-pip \
|
|
||||||
ffmpeg \
|
ffmpeg \
|
||||||
git \
|
git \
|
||||||
git-lfs \
|
git-lfs \
|
||||||
@@ -28,6 +23,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|||||||
|
|
||||||
# Install uv.
|
# Install uv.
|
||||||
COPY --from=ghcr.io/astral-sh/uv:0.6.17 /uv /uvx /usr/local/bin/
|
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
|
WORKDIR /app
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user