diff --git a/devops/containers/ubuntu2204_build.Dockerfile b/devops/containers/ubuntu2204_build.Dockerfile
index fcab26cd900af..d87d205394998 100644
--- a/devops/containers/ubuntu2204_build.Dockerfile
+++ b/devops/containers/ubuntu2204_build.Dockerfile
@@ -12,13 +12,10 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
 COPY scripts/install_build_tools.sh /install.sh
 RUN /install.sh
 
-# Install AMD ROCm
-RUN printf 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600\n' | tee /etc/apt/preferences.d/rocm-pin-600
 RUN apt install -yqq libnuma-dev wget gnupg2 && \
-  wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - && \
-  echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/debian/ ubuntu main' | tee /etc/apt/sources.list.d/rocm.list && \
-  apt update && \
-  apt install -yqq rocm-dev && \
+  wget https://repo.radeon.com/amdgpu-install/6.1/ubuntu/jammy/amdgpu-install_6.1.60100-1_all.deb && \
+  apt install -yqq ./amdgpu-install_6.1.60100-1_all.deb && \
+  yes | amdgpu-install --usecase=rocmdev && \
   apt-get clean && \
   rm -rf /var/lib/apt/lists/*