Skip to content

Commit

Permalink
[CI][HIP] Switch to using 6.1 rocm and make it use a fixed version. (#…
Browse files Browse the repository at this point in the history
…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 <jack.kirk@codeplay.com>
  • Loading branch information
JackAKirk authored Apr 30, 2024
1 parent 4c7baa7 commit a780a8b
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions devops/containers/ubuntu2204_build.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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/*

Expand Down

0 comments on commit a780a8b

Please sign in to comment.