Skip to content

Commit 70309d2

Browse files
committed
NO-JIRA: update ROCm and AMDGPU versions to 6.4.3 in TensorFlow Dockerfile
1 parent 89e3244 commit 70309d2

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

jupyter/rocm/tensorflow/ubi9-python-3.12/Dockerfile.rocm

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,20 +43,21 @@ RUN curl -L https://mirror.openshift.com/pub/openshift-v4/$(uname -m)/clients/oc
4343
rm -f /tmp/openshift-client-linux.tar.gz
4444
# Install the oc client end
4545

46-
########################
46+
#############
4747
# rocm-base #
48-
########################
48+
#############
4949
FROM base AS rocm-base
5050

5151
USER 0
5252
WORKDIR /opt/app-root/bin
5353

5454
# Please keep in sync with ROCm/python3.12 dependent images
55-
ARG ROCM_VERSION=6.4.1
56-
ARG AMDGPU_VERSION=6.4.1
55+
ARG ROCM_VERSION=6.4.3
56+
ARG AMDGPU_VERSION=6.4.3
5757

5858
# Install the ROCm rpms
5959
# ref: https://github.com/ROCm/ROCm-docker/blob/master/dev/Dockerfile-centos-7-complete
60+
# docs: https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/install-methods/package-manager/package-manager-rhel.html#registering-rocm-repositories
6061
# Note: Based on 6.4 above new package mivisionx is a pre-requistes, which bring in more dependent packages
6162
# so we are only installing meta packages of rocm
6263
# ref: https://rocm.docs.amd.com/projects/install-on-linux/en/develop/reference/package-manager-integration.html#packages-in-rocm-programming-models

0 commit comments

Comments
 (0)