Dockerfile.rocm.ubi: bump ROCm to 6.3.1, UBI base to 9.5

opendatahub-io · Jan 10, 2025 · e1d7fc1 · e1d7fc1
1 parent 06d7da8
commit e1d7fc1
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/Dockerfile.rocm.ubi b/Dockerfile.rocm.ubi
@@ -1,5 +1,5 @@
 ## Global Args ##################################################################
-ARG BASE_UBI_IMAGE_TAG=9.4
+ARG BASE_UBI_IMAGE_TAG=9.5-1733767867
 ARG PYTHON_VERSION=3.12
 # Default ROCm ARCHes to build vLLM for.
 ARG PYTORCH_ROCM_ARCH="gfx908;gfx90a;gfx942;gfx1100"
@@ -24,13 +24,13 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 
 FROM base AS rocm_base
-ARG ROCM_VERSION=6.2.3
+ARG ROCM_VERSION=6.3.1
 ARG PYTHON_VERSION
 ARG BASE_UBI_IMAGE_TAG
 
 RUN printf "[amdgpu]\n\
 name=amdgpu\n\
-baseurl=https://repo.radeon.com/amdgpu/${ROCM_VERSION}/rhel/${BASE_UBI_IMAGE_TAG}/main/x86_64/\n\
+baseurl=https://repo.radeon.com/amdgpu/${ROCM_VERSION}/rhel/${BASE_UBI_IMAGE_TAG/-*/}/main/x86_64/\n\
 enabled=1\n\
 priority=50\n\
 gpgcheck=1\n\
@@ -49,8 +49,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
     uv pip install --pre \
         --index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \
-        torch==2.6.0.dev20241122+rocm${version}\
-        torchvision==0.20.0.dev20241122+rocm${version} && \
+        torch==2.6.0.dev20250104+rocm${version}\
+        torchvision==0.22.0.dev20250104+rocm${version} && \
     # Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)
     microdnf install -y libdrm-amdgpu && \
     microdnf clean all
@@ -75,7 +75,7 @@ RUN rpm -ivh https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.
         ccache \
         git \
         # packages required to build vllm
-        amd-smi \
+        amd-smi-lib \
         hipblas-devel \
         hipblaslt-devel \
         hipcc \