Block a user
noteflow-server-rocm-dev (latest)
Published 2026-01-18 16:53:55 +00:00 by vasceannie
Installation
docker pull git.baked.rocks/vasceannie/noteflow-server-rocm-dev:latestsha256:99420bb764dbf87d206bec7ca4d162e839b1ce39c0347609b59ea30d5055a37d
About this package
NoteFlow development server - AMD ROCm GPU build
Image Layers
| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=24.04 |
| ADD file:dafefa97de6dc66a6734ec6f05e58125ce01225cccce3f50662330c252aad518 in / |
| CMD ["/bin/bash"] |
| ARG PYTHON_VERSION |
| ARG PYTORCH_VERSION |
| ARG ROCM_VERSION |
| ARG AMDGPU_VERSION |
| ENV DEBIAN_FRONTEND=noninteractive |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /bin/sh -c echo "tzdata tzdata/Areas select Etc" | debconf-set-selections && echo "tzdata tzdata/Zones/Etc select UTC" | debconf-set-selections # buildkit |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /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 |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /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 |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /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 # buildkit |
| ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /bin/sh -c if [ "$BASE_IMAGE" = "ubuntu:24.04" ]; then wget https://repo.radeon.com/amdgpu-install/$ROCM_VERSION/ubuntu/noble/amdgpu-install_$AMDGPU_VERSION-1_all.deb; else wget https://repo.radeon.com/amdgpu-install/$ROCM_VERSION/ubuntu/jammy/amdgpu-install_$AMDGPU_VERSION-1_all.deb; fi # buildkit |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /bin/sh -c apt install -y ./*.deb # buildkit |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /bin/sh -c amdgpu-install --usecase=rocm -y && rm *.deb # buildkit |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /bin/sh -c set -e && echo "Installing PyTorch via .sh installer for ROCm $ROCM_VERSION, Torch $PYTORCH_VERSION, Python $PYTHON_VERSION" && if [ "$PYTHON_VERSION" = "3.12" ]; then PYTHON_TAG="cp312-cp312"; elif [ "$PYTHON_VERSION" = "3.10" ]; then PYTHON_TAG="cp310-cp310"; else echo "Unsupported Python version: $PYTHON_VERSION"; exit 1; fi && if [ $(echo "$ROCM_VERSION" | grep -o "\." | wc -l) -eq 1 ]; then ROCM_VERSION_FULL="${ROCM_VERSION}.0"; else ROCM_VERSION_FULL="$ROCM_VERSION"; fi && TORCH_SH=$(curl -fsSL https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/.sh-scripts | grep "torch-${PYTORCH_VERSION}%2Brocm${ROCM_VERSION_FULL}.lw" | grep "${PYTHON_TAG}" | grep "\.sh" | sed -n 's/.*href="\([^"]*\).*/\1/p' | head -n1) && if [ -z "$TORCH_SH" ]; then echo "No matching torch .sh installer found for torch-${PYTORCH_VERSION}+rocm${ROCM_VERSION_FULL} (${PYTHON_TAG})"; exit 1; fi && echo "Found torch installer: $TORCH_SH" && mkdir -p /install && curl -fsSL -o /install/torch_installer.sh https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/.sh-scripts/${TORCH_SH} && chmod +x /install/torch_installer.sh && sh /install/torch_installer.sh # buildkit |
| RUN |4 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.7.1 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0.70000 /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 |
| LABEL maintainer=NoteFlow Team |
| LABEL description=NoteFlow with ROCm GPU support for AMD GPUs |
| ENV PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 UV_COMPILE_BYTECODE=1 UV_LINK_MODE=copy PATH=/usr/local/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| COPY /uv /uvx /bin/ # buildkit |
| RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends build-essential pkg-config python3-venv portaudio19-dev libsndfile1 ffmpeg git && rm -rf /var/lib/apt/lists/* # buildkit |
| WORKDIR /workspace |
| COPY pyproject.toml uv.lock* ./ # buildkit |
| COPY README.md ./ # buildkit |
| COPY src ./src/ # buildkit |
| ENV VIRTUAL_ENV=/opt/venv |
| RUN /bin/sh -c uv venv --system-site-packages ${VIRTUAL_ENV} # buildkit |
| ENV PATH=/opt/venv/bin:/usr/local/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| RUN /bin/sh -c uv pip install --python ${VIRTUAL_ENV}/bin/python -e ".[rocm,optional]" # buildkit |
| RUN /bin/sh -c uv pip install --python ${VIRTUAL_ENV}/bin/python watchfiles # buildkit |
| ARG SPACY_MODEL_URL=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 |
| RUN |1 SPACY_MODEL_URL=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 /bin/sh -c uv pip install --python ${VIRTUAL_ENV}/bin/python ${SPACY_MODEL_URL} # buildkit |
| COPY . . # buildkit |
| ENV ROCM_PATH=/opt/rocm HIP_VISIBLE_DEVICES=0 HSA_OVERRIDE_GFX_VERSION= NOTEFLOW_ASR_DEVICE=rocm NOTEFLOW_FEATURE_ROCM_ENABLED=true |
| EXPOSE [50051/tcp] |
| CMD ["python" "scripts/dev_watch_server.py"] |
Labels
| Key | Value |
|---|---|
| ai.noteflow.rocm.version | 7.0 |
| description | NoteFlow with ROCm GPU support for AMD GPUs |
| maintainer | NoteFlow Team |
| org.opencontainers.image.description | NoteFlow development server - AMD ROCm GPU build |
| org.opencontainers.image.ref.name | ubuntu |
| org.opencontainers.image.source | https://github.com/noteflow/noteflow |
| org.opencontainers.image.title | NoteFlow Server Dev (ROCm) |
| org.opencontainers.image.vendor | NoteFlow |
| org.opencontainers.image.version | 24.04 |