ROCm-docker/dev/Dockerfile-manylinux2014_x86_64
2023-06-02 08:42:08 -07:00

15 lines
567 B
Plaintext

FROM quay.io/pypa/manylinux2014_x86_64
LABEL maintainer=jeff.daily@amd.com
#NOTE: manylinux image already has devtoolset installed
ARG ROCM_VERSION=5.4.2
ARG AMDGPU_VERSION=5.4.2
# Install the ROCm rpms
RUN yum clean all
RUN echo -e "[ROCm]\nname=ROCm\nbaseurl=https://repo.radeon.com/rocm/yum/$ROCM_VERSION/main\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/rocm.repo
RUN echo -e "[amdgpu]\nname=amdgpu\nbaseurl=https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/rhel/7.9/main/x86_64\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo
RUN yum install -y rocm-dev