[CI][HIP] Switch to using 6.1 rocm and make it use a fixed version. (#13300)

Previously the version of ROCM used by docker was not fixed and could
change when rocm latest version is updated by AMD, which could lead to
CI problems.

This fixes that and sets the version to 6.1.
Only the rocmdev package is installed which is the minimal required
package for the compiler. This reduces the memory overhead.

---------

Signed-off-by: JackAKirk <[email protected]>

Author: JackAKirk

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/*