chautauqua-worker-rocm-kokoro (sha256:aec9afc59ef89011d6fb0bc4d5d416ca7c1a5c6b3ccf87f6c55105d49e9b85c3)

Published 2026-09-03 02:09:19 -07:00 by admin123

Pull the image from the command line:

docker pull forgejo.tail9dd1e.ts.net/elog08/chautauqua-worker-rocm-kokoro@sha256:aec9afc59ef89011d6fb0bc4d5d416ca7c1a5c6b3ccf87f6c55105d49e9b85c3

Digest

sha256:aec9afc59ef89011d6fb0bc4d5d416ca7c1a5c6b3ccf87f6c55105d49e9b85c3

For more information on the Container registry, see the documentation.

ARG RELEASE

ARG LAUNCHPAD_BUILD_ARCH

LABEL org.opencontainers.image.version=24.04

ADD file:d938ff3d4eee15d8600de84bf85eac6ecd0f20bc92dfe305dafbff0bdc974c0f in /

CMD ["/bin/bash"]

ARG PYTHON_VERSION=3.12

COPY .github/scripts/install_rocm_deps.sh /tmp/ # buildkit

COPY .github/scripts/install_python.sh /tmp/ # buildkit

RUN |1 PYTHON_VERSION=3.12 /bin/sh -c chmod +x /tmp/install_rocm_deps.sh /tmp/install_python.sh && /tmp/install_rocm_deps.sh && /tmp/install_python.sh "${PYTHON_VERSION}" # buildkit

RUN |1 PYTHON_VERSION=3.12 /bin/sh -c python${PYTHON_VERSION} -m venv /opt/venv && /opt/venv/bin/python -m pip install --upgrade pip && /opt/venv/bin/python -m pip install --upgrade setuptools # buildkit

ENV VIRTUAL_ENV=/opt/venv PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

ARG ROCM_VERSION=7.14.0

ARG AMDGPU_FAMILY=device-all

ARG RELEASE_TYPE=nightly

ARG PYTHON_VERSION=3.12

ARG INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch

ARG EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/

ARG PIN_EXPLICIT_VERSIONS=true

ARG TORCH_VERSION=2.13.0+rocm7.14.0

ARG TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.0

ARG TORCHVISION_VERSION=0.28.0+rocm7.14.0

ARG APEX_VERSION

ARG TRITON_VERSION

LABEL image.title=PyTorch + ROCm runtime image (NPI) image.version=7.14.0 npi.python_version=3.12 npi.amdgpu_family=device-all npi.rocm_version=7.14.0

COPY .github/scripts/install_pytorch_wheels.py /tmp/ # buildkit

COPY .github/scripts/detect_device_arches.py /tmp/ # buildkit

RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all RELEASE_TYPE=nightly INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/ PIN_EXPLICIT_VERSIONS=true TORCH_VERSION=2.13.0+rocm7.14.0 TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.0 TORCHVISION_VERSION=0.28.0+rocm7.14.0 APEX_VERSION= TRITON_VERSION= /bin/sh -c set -eu; PIN_EXPLICIT_FLAG=""; if [ "${PIN_EXPLICIT_VERSIONS}" = "true" ]; then PIN_EXPLICIT_FLAG="--pin-explicit-versions"; fi; /opt/venv/bin/python /tmp/install_pytorch_wheels.py --no-break-system-packages --skip-verify --device-targets "${AMDGPU_FAMILY}" --index-url "${INDEX_URL}" ${EXTRA_INDEX_URL:+--extra-index-url "${EXTRA_INDEX_URL}"} --amdgpu-family "${AMDGPU_FAMILY}" ${ROCM_VERSION:+--rocm-version "${ROCM_VERSION}"} ${PIN_EXPLICIT_FLAG} ${TORCH_VERSION:+--torch-version "${TORCH_VERSION}"} ${TORCHAUDIO_VERSION:+--torchaudio-version "${TORCHAUDIO_VERSION}"} ${TORCHVISION_VERSION:+--torchvision-version "${TORCHVISION_VERSION}"} ${APEX_VERSION:+--apex-version "${APEX_VERSION}"} ${TRITON_VERSION:+--triton-version "${TRITON_VERSION}"} # buildkit

ARG PYTHON_VERSION=3.12

ARG INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch

ARG EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/

ARG AMDGPU_FAMILY=device-all

ARG ROCM_VERSION=7.14.0

ARG PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx950

ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_SOURCE_DIR=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_BIN=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin ROCM_CMAKE=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/cmake ROCM_CMAKE_PREFIX=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/cmake BUILD_ENVIRONMENT=rocm PYTORCH_SRC_PATH=/workspace PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx950 USE_MSLK=0

ENV CMAKE_PREFIX_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/cmake HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode ROCM_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode ROCM_SYSDEPS_INCLUDE=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/include CPLUS_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/include C_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/include PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib/pkgconfig LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all RELEASE_TYPE=nightly INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/ PIN_EXPLICIT_VERSIONS=true TORCH_VERSION=2.13.0+rocm7.14.0 TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.0 TORCHVISION_VERSION=0.28.0+rocm7.14.0 APEX_VERSION= TRITON_VERSION= PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx950 /bin/sh -c ROCM_INDEX_URL="${INDEX_URL%/}/${AMDGPU_FAMILY}/" && if [ "${AMDGPU_FAMILY}" = "all" ] || [ "${AMDGPU_FAMILY}" = "device-all" ] || [ "${AMDGPU_FAMILY}" = "multi-arch" ]; then ROCM_INDEX_URL="${INDEX_URL%/}/"; fi && EXTRA_INDEX_ARGS="" && if [ -n "${EXTRA_INDEX_URL}" ]; then EXTRA_INDEX_ARGS="--extra-index-url ${EXTRA_INDEX_URL}"; fi && if [ -n "${ROCM_VERSION}" ]; then /opt/venv/bin/python -m pip install --index-url "${ROCM_INDEX_URL}" ${EXTRA_INDEX_ARGS} "rocm[devel]==${ROCM_VERSION}"; else /opt/venv/bin/python -m pip install --index-url "${ROCM_INDEX_URL}" ${EXTRA_INDEX_ARGS} "rocm[devel]"; fi # buildkit

COPY pytorch /workspace/pytorch # buildkit

RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all RELEASE_TYPE=nightly INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/ PIN_EXPLICIT_VERSIONS=true TORCH_VERSION=2.13.0+rocm7.14.0 TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.0 TORCHVISION_VERSION=0.28.0+rocm7.14.0 APEX_VERSION= TRITON_VERSION= PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx950 /bin/sh -c if [ -f /workspace/pytorch/.ci/docker/requirements-ci.txt ]; then /opt/venv/bin/python -m pip install -r /workspace/pytorch/.ci/docker/requirements-ci.txt; fi && if [ -f /workspace/pytorch/requirements.txt ]; then /opt/venv/bin/python -m pip install -r /workspace/pytorch/requirements.txt; fi && /opt/venv/bin/python -m pip install pytest pytest-rerunfailures pytest-shard hypothesis scipy ninja # buildkit

RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all RELEASE_TYPE=nightly INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/ PIN_EXPLICIT_VERSIONS=true TORCH_VERSION=2.13.0+rocm7.14.0 TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.0 TORCHVISION_VERSION=0.28.0+rocm7.14.0 APEX_VERSION= TRITON_VERSION= PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx950 /bin/sh -c /opt/venv/bin/python <<'PYEOF' import os import torch print('torch', torch.__version__) print('ROCm/HIP', torch.version.hip) for name in ('ROCM_HOME', 'CC', 'CXX', 'CMAKE_PREFIX_PATH'): print(f'{name}={os.environ.get(name, "NOT SET")}') for mod in ['torchaudio', 'torchvision', 'apex', 'triton']: try: module = __import__(mod) print(f'{mod} {module.__version__}') except Exception as exc: print(f'{mod}: skipped ({exc})') PYEOF # buildkit

WORKDIR /

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 fonts-sil-abyssinica git && rm -rf /var/lib/apt/lists/* # buildkit

WORKDIR /app

ENV UV_PROJECT_ENVIRONMENT=/opt/venv

ENV PATH=/opt/venv/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/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.45

ENV CHAUTAUQUA_BUILD_ID=v0.1.45

RUN |2 MODEL_EXTRA=kokoro-gpu CHAUTAUQUA_BUILD_ID=v0.1.45 /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.45 /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.45 /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.45 /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

Key	Value
image.title	PyTorch + ROCm runtime image (NPI)
image.version	7.14.0
npi.amdgpu_family	device-all
npi.python_version	3.12
npi.rocm_version	7.14.0
org.opencontainers.image.version	24.04

Details

Container

2026-09-03 02:09:19 -07:00

OCI / Docker

linux/amd64

28 GiB

Versions (30) View all

0.1

2026-09-03

latest

2026-09-03

0.1.45

2026-09-03

buildcache

2026-09-03

sha-89c0e46f9d67

2026-09-03

chautauqua-worker-rocm-kokoro (sha256:aec9afc59ef89011d6fb0bc4d5d416ca7c1a5c6b3ccf87f6c55105d49e9b85c3)

Installation

Image layers

Labels