File tree Expand file tree Collapse file tree 5 files changed +30
-15
lines changed
tensorflow/ubi9-python-3.11
rocm-pytorch/ubi9-python-3.11
rocm-tensorflow/ubi9-python-3.11 Expand file tree Collapse file tree 5 files changed +30
-15
lines changed Original file line number Diff line number Diff line change @@ -32,11 +32,14 @@ USER 0
32
32
WORKDIR /opt/app-root/bin
33
33
34
34
# Please keep in sync with ROCm/python3.11 dependent images
35
- ARG ROCM_VERSION=6.1
36
- ARG AMDGPU_VERSION=6.1
35
+ ARG ROCM_VERSION=6.2.4
36
+ ARG AMDGPU_VERSION=6.2.4
37
37
38
38
# Install the ROCm rpms
39
39
# ref: https://github.com/ROCm/ROCm-docker/blob/master/dev/Dockerfile-centos-7-complete
40
+ # Note: Based on 6.2 above new package mivisionx is a pre-requistes, which bring in more dependent packages
41
+ # so we are only installing meta packages of rocm
42
+ # ref: https://rocm.docs.amd.com/projects/install-on-linux/en/develop/reference/package-manager-integration.html#packages-in-rocm-programming-models
40
43
RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
41
44
echo "name=ROCm" >> /etc/yum.repos.d/rocm.repo && \
42
45
echo "baseurl=https://repo.radeon.com/rocm/rhel9/$ROCM_VERSION/main" >> /etc/yum.repos.d/rocm.repo && \
@@ -47,7 +50,7 @@ RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
47
50
echo "baseurl=https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/rhel/9.4/main/x86_64" >> /etc/yum.repos.d/amdgpu.repo && \
48
51
echo "enabled=1" >> /etc/yum.repos.d/amdgpu.repo && \
49
52
echo "gpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo && \
50
- yum install -y rocm && \
53
+ yum install -y rocm-developer-tools rocm-ml-sdk rocm-opencl-sdk rocm-openmp-sdk rocm-utils && \
51
54
yum clean all && rm -rf /var/cache/yum
52
55
53
56
# Restore notebook user workspace
Original file line number Diff line number Diff line change @@ -32,11 +32,14 @@ USER 0
32
32
WORKDIR /opt/app-root/bin
33
33
34
34
# Please keep in sync with ROCm/python3.11 dependent images
35
- ARG ROCM_VERSION=6.1
36
- ARG AMDGPU_VERSION=6.1
35
+ ARG ROCM_VERSION=6.2.4
36
+ ARG AMDGPU_VERSION=6.2.4
37
37
38
38
# Install the ROCm rpms
39
39
# ref: https://github.com/ROCm/ROCm-docker/blob/master/dev/Dockerfile-centos-7-complete
40
+ # Note: Based on 6.2 above new package mivisionx is a pre-requistes, which bring in more dependent packages
41
+ # so we are only installing meta packages of rocm
42
+ # ref: https://rocm.docs.amd.com/projects/install-on-linux/en/develop/reference/package-manager-integration.html#packages-in-rocm-programming-models
40
43
RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
41
44
echo "name=ROCm" >> /etc/yum.repos.d/rocm.repo && \
42
45
echo "baseurl=https://repo.radeon.com/rocm/rhel9/$ROCM_VERSION/main" >> /etc/yum.repos.d/rocm.repo && \
@@ -47,7 +50,7 @@ RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
47
50
echo "baseurl=https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/rhel/9.4/main/x86_64" >> /etc/yum.repos.d/amdgpu.repo && \
48
51
echo "enabled=1" >> /etc/yum.repos.d/amdgpu.repo && \
49
52
echo "gpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo && \
50
- yum install -y rocm && \
53
+ yum install -y rocm-developer-tools rocm-ml-sdk rocm-opencl-sdk rocm-openmp-sdk rocm-utils && \
51
54
yum clean all && rm -rf /var/cache/yum
52
55
53
56
# Restore notebook user workspace
Original file line number Diff line number Diff line change @@ -32,11 +32,14 @@ USER 0
32
32
WORKDIR /opt/app-root/bin
33
33
34
34
# Please keep in sync with ROCm/python3.11 dependent images
35
- ARG ROCM_VERSION=6.1
36
- ARG AMDGPU_VERSION=6.1
35
+ ARG ROCM_VERSION=6.2.4
36
+ ARG AMDGPU_VERSION=6.2.4
37
37
38
38
# Install the ROCm rpms
39
39
# ref: https://github.com/ROCm/ROCm-docker/blob/master/dev/Dockerfile-centos-7-complete
40
+ # Note: Based on 6.2 above new package mivisionx is a pre-requistes, which bring in more dependent packages
41
+ # so we are only installing meta packages of rocm
42
+ # ref: https://rocm.docs.amd.com/projects/install-on-linux/en/develop/reference/package-manager-integration.html#packages-in-rocm-programming-models
40
43
RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
41
44
echo "name=ROCm" >> /etc/yum.repos.d/rocm.repo && \
42
45
echo "baseurl=https://repo.radeon.com/rocm/rhel9/$ROCM_VERSION/main" >> /etc/yum.repos.d/rocm.repo && \
@@ -47,7 +50,7 @@ RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
47
50
echo "baseurl=https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/rhel/9.4/main/x86_64" >> /etc/yum.repos.d/amdgpu.repo && \
48
51
echo "enabled=1" >> /etc/yum.repos.d/amdgpu.repo && \
49
52
echo "gpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo && \
50
- yum install -y rocm && \
53
+ yum install -y rocm-developer-tools rocm-ml-sdk rocm-opencl-sdk rocm-openmp-sdk rocm-utils && \
51
54
yum clean all && rm -rf /var/cache/yum
52
55
53
56
# Restore notebook user workspace
Original file line number Diff line number Diff line change @@ -32,11 +32,14 @@ USER 0
32
32
WORKDIR /opt/app-root/bin
33
33
34
34
# Please keep in sync with ROCm/python3.11 dependent images
35
- ARG ROCM_VERSION=6.1
36
- ARG AMDGPU_VERSION=6.1
35
+ ARG ROCM_VERSION=6.2.4
36
+ ARG AMDGPU_VERSION=6.2.4
37
37
38
38
# Install the ROCm rpms
39
39
# ref: https://github.com/ROCm/ROCm-docker/blob/master/dev/Dockerfile-centos-7-complete
40
+ # Note: Based on 6.2 above new package mivisionx is a pre-requistes, which bring in more dependent packages
41
+ # so we are only installing meta packages of rocm
42
+ # ref: https://rocm.docs.amd.com/projects/install-on-linux/en/develop/reference/package-manager-integration.html#packages-in-rocm-programming-models
40
43
RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
41
44
echo "name=ROCm" >> /etc/yum.repos.d/rocm.repo && \
42
45
echo "baseurl=https://repo.radeon.com/rocm/rhel9/$ROCM_VERSION/main" >> /etc/yum.repos.d/rocm.repo && \
@@ -47,7 +50,7 @@ RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
47
50
echo "baseurl=https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/rhel/9.4/main/x86_64" >> /etc/yum.repos.d/amdgpu.repo && \
48
51
echo "enabled=1" >> /etc/yum.repos.d/amdgpu.repo && \
49
52
echo "gpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo && \
50
- yum install -y rocm && \
53
+ yum install -y rocm-developer-tools rocm-ml-sdk rocm-opencl-sdk rocm-openmp-sdk rocm-utils && \
51
54
yum clean all && rm -rf /var/cache/yum
52
55
53
56
# Restore notebook user workspace
Original file line number Diff line number Diff line change @@ -32,11 +32,14 @@ USER 0
32
32
WORKDIR /opt/app-root/bin
33
33
34
34
# Please keep in sync with ROCm/python3.11 dependent images
35
- ARG ROCM_VERSION=6.1
36
- ARG AMDGPU_VERSION=6.1
35
+ ARG ROCM_VERSION=6.2.4
36
+ ARG AMDGPU_VERSION=6.2.4
37
37
38
38
# Install the ROCm rpms
39
39
# ref: https://github.com/ROCm/ROCm-docker/blob/master/dev/Dockerfile-centos-7-complete
40
+ # Note: Based on 6.2 above new package mivisionx is a pre-requistes, which bring in more dependent packages
41
+ # so we are only installing meta packages of rocm
42
+ # ref: https://rocm.docs.amd.com/projects/install-on-linux/en/develop/reference/package-manager-integration.html#packages-in-rocm-programming-models
40
43
RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
41
44
echo "name=ROCm" >> /etc/yum.repos.d/rocm.repo && \
42
45
echo "baseurl=https://repo.radeon.com/rocm/rhel9/$ROCM_VERSION/main" >> /etc/yum.repos.d/rocm.repo && \
@@ -47,7 +50,7 @@ RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
47
50
echo "baseurl=https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/rhel/9.4/main/x86_64" >> /etc/yum.repos.d/amdgpu.repo && \
48
51
echo "enabled=1" >> /etc/yum.repos.d/amdgpu.repo && \
49
52
echo "gpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo && \
50
- yum install -y rocm && \
53
+ yum install -y rocm-developer-tools rocm-ml-sdk rocm-opencl-sdk rocm-openmp-sdk rocm-utils && \
51
54
yum clean all && rm -rf /var/cache/yum
52
55
53
56
# Restore notebook user workspace
You can’t perform that action at this time.
0 commit comments