| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.version=24.04 |
| ADD file:8ce1caf246e7c778bca84c516d02fd4e83766bb2c530a0fffa8a351b560a2728 in / |
| CMD ["/bin/bash"] |
| ARG BASE_IMAGE=ubuntu:24.04 |
| ARG PYTHON_VERSION=3.12 |
| ARG PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 |
| ARG TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 |
| ARG TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f |
| ARG TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 |
| ARG APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 |
| ARG ROCM_VERSION=7.2.4 |
| ARG UBUNTU_VERSION=24.04 |
| ARG PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ |
| LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0+rocm7.2.4.lw.git3d3aa833 |
| ENV DEBIAN_FRONTEND=noninteractive |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c echo "tzdata tzdata/Areas select Etc" | debconf-set-selections && echo "tzdata tzdata/Zones/Etc select UTC" | debconf-set-selections # buildkit |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c apt update && apt install -y wget software-properties-common curl rsync dialog git gnupg # buildkit |
| ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/opt/rocm/lib |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c add-apt-repository ppa:deadsnakes/ppa && apt update && apt install -y python${PYTHON_VERSION} python3-pip python3-venv && if [ "${PYTHON_VERSION}" = "3.10" ]; then apt install -y --no-install-recommends python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-distutils python${PYTHON_VERSION}-venv; elif [ "${PYTHON_VERSION}" = "3.12" ]; then apt install -y --no-install-recommends python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv; else echo "Only PYTHON_VERSION 3.10 or 3.12 is supported" && exit 1; fi && apt clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c python${PYTHON_VERSION} -m venv /opt/venv && . /opt/venv/bin/activate && /opt/venv/bin/python -m pip install --upgrade pip setuptools && /opt/venv/bin/python -m pip install hypothesis scipy ninja # buildkit |
| ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c set -eux; mkdir -p --mode=0755 /etc/apt/keyrings; wget -qO- https://repo.radeon.com/rocm/rocm.gpg.key | gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null; if [ "$PYTHON_VERSION" = "3.12" ]; then CODENAME=noble; elif [ "$PYTHON_VERSION" = "3.10" ]; then CODENAME=jammy; else echo "Only PYTHON_VERSION 3.10 or 3.12 is supported" && exit 1; fi; printf 'deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/%s %s main\n' "${ROCM_VERSION}" "${CODENAME}" > /etc/apt/sources.list.d/rocm.list; printf '%s\n' 'Package: *' 'Pin: release o=repo.radeon.com' 'Pin-Priority: 600' > /etc/apt/preferences.d/rocm-pin-600; apt-get update; apt-get install -y --no-install-recommends rocm python3-setuptools python3-wheel; apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c set -e && WHEEL_INDEX="${PYTORCH_WHEEL_INDEX:-https://repo.radeon.com/rocm/manylinux/rocm-rel-${ROCM_VERSION}/}" && echo "Installing from $WHEEL_INDEX Python $PYTHON_VERSION" && pip install --no-cache-dir --find-links "${WHEEL_INDEX}" --prefer-binary "torch==${PYTORCH_VERSION}" "torchvision==${TORCHVISION_VERSION}" "torchaudio==${TORCHAUDIO_VERSION}" "triton==${TRITON_VERSION}" && pip install --no-cache-dir --find-links "${WHEEL_INDEX}" --prefer-binary "apex==${APEX_VERSION}" && for pkg in torch torchvision torchaudio triton apex; do pip show "$pkg" >/dev/null 2>&1 || { echo "ERROR: $pkg missing"; exit 1; }; done && python -c "import torch, torchvision, torchaudio; tv=torch.__version__; vv=torchvision.__version__; av=torchaudio.__version__; assert '+rocm' in tv, f'Expected ROCm torch wheel, got: {tv}'; assert '+rocm' in vv, f'Expected ROCm torchvision wheel, got: {vv}'; assert '+rocm' in av, f'Expected ROCm torchaudio wheel, got: {av}'; print('torch:', tv, 'hip:', torch.version.hip); print('torchvision:', vv); print('torchaudio:', av)" # buildkit |
| RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833 TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f59 TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f TRITON_VERSION=3.6.0+rocm7.2.4.git4ed88892 APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd5 ROCM_VERSION=7.2.4 UBUNTU_VERSION=24.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/ /bin/sh -c git clone https://github.com/ROCm/pytorch-micro-benchmarking /var/lib/jenkins/pytorch-micro-benchmarking && ln -sf /usr/bin/python3 /usr/bin/python # buildkit |
| COPY /uv /usr/local/bin/uv # buildkit |
| RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends ffmpeg espeak-ng libsndfile1 sox fontconfig fonts-dejavu-core git && rm -rf /var/lib/apt/lists/* # buildkit |
| WORKDIR /app |
| ENV UV_PROJECT_ENVIRONMENT=/opt/venv |
| ENV PATH=/opt/venv/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PYTHONPATH=/app |
| COPY pyproject.toml uv.lock ./ # buildkit |
| ARG MODEL_EXTRA=kokoro-gpu |
| ARG CHAUTAUQUA_BUILD_ID=v0.1.30 |
| ENV CHAUTAUQUA_BUILD_ID=v0.1.30 |
| RUN |2 MODEL_EXTRA=kokoro-gpu CHAUTAUQUA_BUILD_ID=v0.1.30 /bin/sh -c uv sync --inexact --extra temporal --extra ingest --extra transform --extra storage-minio --extra kokoro-gpu --extra video --extra fx --no-install-package torch --no-install-package torchaudio --no-install-package torchvision --no-install-project --frozen # buildkit |
| RUN |2 MODEL_EXTRA=kokoro-gpu CHAUTAUQUA_BUILD_ID=v0.1.30 /bin/sh -c uv pip install --python /opt/venv/bin/python --no-deps https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0-py3-none-any.whl # buildkit |
| RUN |2 MODEL_EXTRA=kokoro-gpu CHAUTAUQUA_BUILD_ID=v0.1.30 /bin/sh -c /opt/venv/bin/python -c "import torch; print(f'torch={torch.__version__} hip={torch.version.hip} cuda_avail={torch.cuda.is_available()}')" || echo "WARN: build-time torch import failed (SIGILL on a CPU-only/edge builder is expected); torch is validated at runtime on the ROCm GPU host" # buildkit |
| RUN |2 MODEL_EXTRA=kokoro-gpu CHAUTAUQUA_BUILD_ID=v0.1.30 /bin/sh -c /opt/venv/bin/python -c "import spacy; spacy.load('en_core_web_sm'); print('en_core_web_sm OK')" || echo "WARN: build-time spacy import failed (SIGILL on a CPU-only/edge builder is expected); validated at runtime on the ROCm GPU host" # buildkit |
| COPY chautauqua /app/chautauqua # buildkit |
| COPY cuda_tts /app/cuda_tts # buildkit |
| COPY mlx_tts /app/mlx_tts # buildkit |
| COPY gemini_tts /app/gemini_tts # buildkit |
| COPY scripts /app/scripts # buildkit |
