Skip to content

Commit 9fe0af4

Browse files
upgrade AMD Ray runtime image to ROCm v6.2.4
1 parent 9cb3de8 commit 9fe0af4

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

.tekton/ray-rocm-push.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@ spec:
2424
- name: revision
2525
value: '{{revision}}'
2626
- name: output-image
27-
value: quay.io/modh/ray:2.35.0-py311-rocm61
27+
value: quay.io/modh/ray:2.35.0-py311-rocm62
2828
- name: additional-tag
29-
value: 2.35.0-py311-rocm61-{{revision}}
29+
value: 2.35.0-py311-rocm62-{{revision}}
3030
- name: dockerfile
3131
value: Dockerfile
3232
- name: path-context

images/runtime/ray/rocm/Dockerfile

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
ARG PYTHON_VERSION=311
2-
ARG IMAGE_TAG=1-77.1726664316
2+
ARG IMAGE_TAG=9.5-1737537151
33

44
FROM registry.access.redhat.com/ubi9/python-${PYTHON_VERSION}:${IMAGE_TAG}
55

6-
LABEL name="ray-ubi9-py311-rocm61" \
6+
LABEL name="ray-ubi9-py311-rocm62" \
77
summary="ROCm Python 3.11 image based on UBI9 for Ray" \
88
description="ROCm Python 3.11 image based on UBI9 for Ray" \
99
io.k8s.display-name="ROCm Python 3.11 base image for Ray" \
@@ -14,8 +14,8 @@ LABEL name="ray-ubi9-py311-rocm61" \
1414
USER 0
1515
WORKDIR /opt/app-root/bin
1616

17-
ARG ROCM_VERSION=6.1.2
18-
ARG AMDGPU_VERSION=6.1.2
17+
ARG ROCM_VERSION=6.2.4
18+
ARG AMDGPU_VERSION=6.2.4
1919

2020
RUN <<EOF
2121
cat <<EOD > /etc/yum.repos.d/rocm.repo
@@ -37,7 +37,7 @@ gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key
3737
EOD
3838
EOF
3939

40-
RUN yum -y install rocm && yum clean all && rm -rf /var/cache/yum
40+
RUN yum install -y rocm-developer-tools rocm-ml-sdk rocm-opencl-sdk rocm-openmp-sdk rocm-utils && yum clean all && rm -rf /var/cache/yum
4141

4242
# Install Python packages
4343

0 commit comments

Comments
 (0)