11ARG UBUNTU_VERSION=24.04
22
33# This needs to generally match the container host's environment.
4- ARG ROCM_VERSION=7.2
5- ARG AMDGPU_VERSION=7.2
4+ ARG ROCM_VERSION=7.2.1
5+ ARG AMDGPU_VERSION=7.2.1
66
77# Target the ROCm build image
88ARG BASE_ROCM_DEV_CONTAINER=rocm/dev-ubuntu-${UBUNTU_VERSION}:${ROCM_VERSION}-complete
@@ -12,11 +12,11 @@ FROM ${BASE_ROCM_DEV_CONTAINER} AS build
1212
1313# Unless otherwise specified, we make a fat build.
1414# This is mostly tied to rocBLAS supported archs.
15- # check https://rocm.docs.amd.com/projects/install-on-linux/en/docs-7.2.0 /reference/system-requirements.html
15+ # check https://rocm.docs.amd.com/projects/install-on-linux/en/docs-7.2.1 /reference/system-requirements.html
1616# check https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/compatibility/compatibilityrad/native_linux/native_linux_compatibility.html
1717# check https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/compatibility/compatibilityryz/native_linux/native_linux_compatibility.html
1818
19- ARG ROCM_DOCKER_ARCH='gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1151;gfx1150;gfx1200;gfx1201'
19+ ARG ROCM_DOCKER_ARCH='gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1102; gfx1151;gfx1150;gfx1200;gfx1201'
2020
2121# Set ROCm architectures
2222ENV AMDGPU_TARGETS=${ROCM_DOCKER_ARCH}
@@ -58,7 +58,7 @@ RUN mkdir -p /app/full \
5858FROM ${BASE_ROCM_DEV_CONTAINER} AS base
5959
6060RUN apt-get update \
61- && apt-get install -y libgomp1 curl\
61+ && apt-get install -y libgomp1 curl \
6262 && apt autoremove -y \
6363 && apt clean -y \
6464 && rm -rf /tmp/* /var/tmp/* \
@@ -79,7 +79,7 @@ RUN apt-get update \
7979 git \
8080 python3-pip \
8181 python3 \
82- python3-wheel\
82+ python3-wheel \
8383 && pip install --break-system-packages --upgrade setuptools \
8484 && pip install --break-system-packages -r requirements.txt \
8585 && apt autoremove -y \
0 commit comments