| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:52c0e467fa2e92f101018df01a0ff56580c752b7553fbe6df88e16b02af6d4ee in / |
| CMD ["/bin/bash"] |
| ARG BASE_IMAGE=ubuntu:22.04 |
| ARG PYTHON_VERSION=3.11 |
| ARG PYTORCH_VERSION=2.10.0 |
| ARG TORCHVISION_VERSION=0.25.0 |
| ARG TORCHAUDIO_VERSION=2.10.0 |
| ARG TRITON_VERSION=3.6.0 |
| ARG APEX_VERSION=1.10.0 |
| ARG ROCM_VERSION=7.1.1 |
| ARG AMDGPU_VERSION=7.1.1.70101 |
| ARG UBUNTU_VERSION=22.04 |
| ARG PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ |
| LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.11_pytorch_release_2.10.0 |
| ENV DEBIAN_FRONTEND=noninteractive |
| RUN |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /bin/sh -c echo "tzdata tzdata/Areas select Etc" | debconf-set-selections && echo "tzdata tzdata/Zones/Etc select UTC" | debconf-set-selections # buildkit |
| RUN |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /bin/sh -c apt update && apt install -y wget software-properties-common curl rsync dialog git # buildkit |
| ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/opt/rocm/lib |
| RUN |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /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.8" ] || [ "${PYTHON_VERSION}" = "3.9" ] || [ "${PYTHON_VERSION}" = "3.10" ] || [ "${PYTHON_VERSION}" = "3.11" ]; then apt install -y --no-install-recommends python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-distutils python${PYTHON_VERSION}-venv; else apt install -y --no-install-recommends python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv; fi && apt clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /bin/sh -c python${PYTHON_VERSION} -m venv /opt/venv && . /opt/venv/bin/activate && /opt/venv/bin/python -m pip install --upgrade pip && /opt/venv/bin/python -m pip install --upgrade 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 |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /bin/sh -c cd /tmp && if [ "$PYTHON_VERSION" = "3.12" ] || [ "$PYTHON_VERSION" = "3.13" ]; then DISTRO="noble"; elif [ "$PYTHON_VERSION" = "3.11" ] || [ "$PYTHON_VERSION" = "3.10" ] || [ "$PYTHON_VERSION" = "3.9" ] || [ "$PYTHON_VERSION" = "3.8" ]; then DISTRO="jammy"; else echo "Unsupported PYTHON_VERSION: $PYTHON_VERSION" && exit 1; fi && wget https://repo.radeon.com/amdgpu-install/${ROCM_VERSION}/ubuntu/${DISTRO}/amdgpu-install_${AMDGPU_VERSION}-1_all.deb; apt install -y ./*.deb && amdgpu-install --usecase=rocm -y && rm -f ./*.deb && apt clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /bin/sh -c set -e && WHEEL_INDEX="${PYTORCH_WHEEL_INDEX:-https://repo.radeon.com/rocm/manylinux/rocm-rel-${ROCM_VERSION}/}" && echo "Installing PyTorch ROCm from $WHEEL_INDEX for Python $PYTHON_VERSION" && pip install --find-links "${WHEEL_INDEX}" --prefer-binary "torch~=${PYTORCH_VERSION}" "torchvision~=${TORCHVISION_VERSION}" "torchaudio~=${TORCHAUDIO_VERSION}" "triton~=${TRITON_VERSION}" && pip install --find-links "${WHEEL_INDEX}" --prefer-binary "apex~=${APEX_VERSION}" 2>/dev/null || true && for pkg in torch torchvision torchaudio triton apex; do pip show "$pkg" >/dev/null 2>&1 || { echo "ERROR: Package $pkg not installed"; exit 1; }; done # buildkit |
| RUN |11 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.10.0 TORCHVISION_VERSION=0.25.0 TORCHAUDIO_VERSION=2.10.0 TRITON_VERSION=3.6.0 APEX_VERSION=1.10.0 ROCM_VERSION=7.1.1 AMDGPU_VERSION=7.1.1.70101 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/ /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 |
| WORKDIR /app |
| RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends build-essential && rm -rf /var/lib/apt/lists/* # buildkit |
| COPY requirements.txt . # buildkit |
| RUN /bin/sh -c pip install --no-cache-dir -r requirements.txt # buildkit |
| COPY wyoming_glados/ ./wyoming_glados/ # buildkit |
| COPY download_model.py . # buildkit |
| COPY entrypoint.sh . # buildkit |
| RUN /bin/sh -c chmod +x entrypoint.sh # buildkit |
| RUN /bin/sh -c mkdir -p /data # buildkit |
| EXPOSE [10200/tcp] |
| ENTRYPOINT ["/app/entrypoint.sh"] |
