Jānis janhouse

search-platform-embedder-gpu (dab0d4af63b759c055785fcc8877b35ca31d89ee)

Published 2026-04-29 21:09:19 +00:00 by janhouse

Installation

docker pull git.janhouse.lv/janhouse/search-platform-embedder-gpu:dab0d4af63b759c055785fcc8877b35ca31d89ee
sha256:0c08439c11e304e2e7463f900908b1bb8d7f8c2bfbdf7da02643a286011bb4a7

Image Layers

ARG RELEASE
ARG LAUNCHPAD_BUILD_ARCH
LABEL org.opencontainers.image.version=22.04
ADD file:6d6bdec36f3282e8506d4ebfcecc427191e59c9cf197a51a9e5787e7490eb0d6 in /
CMD ["/bin/bash"]
ARG BASE_IMAGE=ubuntu:22.04
ARG PYTHON_VERSION=3.10
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.2.2
ARG UBUNTU_VERSION=22.04
ARG PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/
LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.2.2_ubuntu22.04_py3.10_pytorch_release_2.10.0
ENV DEBIAN_FRONTEND=noninteractive
RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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 # buildkit
RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 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.2.2 UBUNTU_VERSION=22.04 PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/ /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
COPY docker/dockerfiles/embedder/requirements.txt . # buildkit
RUN /bin/sh -c pip install --no-cache-dir -r requirements.txt # buildkit
RUN /bin/sh -c python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/LaBSE')" # buildkit
EXPOSE [8000/tcp]
WORKDIR /app
COPY docker/dockerfiles/embedder/main.py . # buildkit
CMD ["uvicorn" "main:app" "--host" "0.0.0.0" "--port" "8000"]

Labels

Key Value
org.opencontainers.image.ref.name rocm/pytorch:rocm7.2.2_ubuntu22.04_py3.10_pytorch_release_2.10.0
org.opencontainers.image.version 22.04
Details
Container
2026-04-29 21:09:19 +00:00
3
OCI / Docker
linux/amd64
11 GiB
Versions (5) View all