diff --git a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml index bf890944e7..6a38aeaf6b 100644 --- a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml +++ b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml @@ -11,42 +11,6 @@ resources: stages: - stage: Python_Packaging jobs: - - template: templates/rocm.yml - parameters: - PythonVersion: '3.8' - RocmVersion: '5.2.3' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.9' - RocmVersion: '5.2.3' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.10' - RocmVersion: '5.2.3' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.8' - RocmVersion: '5.3.2' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.9' - RocmVersion: '5.3.2' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.10' - RocmVersion: '5.3.2' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.8' - RocmVersion: '5.4' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.9' - RocmVersion: '5.4' - - template: templates/rocm.yml - parameters: - PythonVersion: '3.10' - RocmVersion: '5.4' - template: templates/rocm.yml parameters: PythonVersion: '3.8' diff --git a/tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_rocm b/tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_rocm index d6fa029a12..78adeafa95 100644 --- a/tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_rocm +++ b/tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_rocm @@ -15,11 +15,7 @@ RUN yum --enablerepo=extras install -y epel-release 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 -# After ROCm5.2.3, amdgpu_version use the same version as rocm_version -RUN if [ "$ROCM_VERSION" = "5.2.3" ] ; \ - then echo -e "[amdgpu]\nname=amdgpu\nbaseurl=https://repo.radeon.com/amdgpu/22.20.3/rhel/7.9/main/x86_64\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo; \ - else echo -e "[amdgpu]\nname=amdgpu\nbaseurl=https://repo.radeon.com/amdgpu/$ROCM_VERSION/rhel/7.9/main/x86_64\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo; \ - fi +RUN echo -e "[amdgpu]\nname=amdgpu\nbaseurl=https://repo.radeon.com/amdgpu/$ROCM_VERSION/rhel/7.9/main/x86_64\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo RUN yum install -y rocm-dev