mikronova/kyokkou

docker/Dockerfile.rocm

FROM kyokkou-base-noble

# ROCm/HIP — AMD GPU Cycles
# Requires /dev/kfd and /dev/dri passed through via bollard (no special runtime needed)
RUN apt-get update && apt-get install -y --no-install-recommends gnupg \
 && rm -rf /var/lib/apt/lists/* \
 && wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key \
      | gpg --dearmor -o /etc/apt/keyrings/rocm.gpg \
 && echo 'deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg trusted=yes] https://repo.radeon.com/rocm/apt/7.2.2 noble main' \
      > /etc/apt/sources.list.d/rocm.list \
 && printf 'Package: *\nPin: origin repo.radeon.com\nPin-Priority: 1001\n' \
      > /etc/apt/preferences.d/rocm \
 && apt-get update && apt-get install -y --no-install-recommends \
    rocm-hip-runtime \
    rocm-device-libs \
 && rm -rf /var/lib/apt/lists/*

ENV ROCM_PATH=/opt/rocm
ENV PATH="${ROCM_PATH}/bin:${PATH}"
ENV LD_LIBRARY_PATH="${ROCM_PATH}/lib:${LD_LIBRARY_PATH}"