Skip to content

Commit 0300ac9

Browse files
authored
[CI][HIP] Use apt to install rocm6.1.1 (#13771)
Switch to using apt packages instead of amdgpu-install. This uses c++-11-dev libraries so hopefully avoids the problems caused by #13752 Signed-off-by: JackAKirk <[email protected]>
1 parent b758dda commit 0300ac9

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

devops/containers/ubuntu2204_build.Dockerfile

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,22 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
1212
COPY scripts/install_build_tools.sh /install.sh
1313
RUN /install.sh
1414

15-
RUN apt install -yqq libnuma-dev wget gnupg2 && \
16-
apt-get install -yqq libstdc++-12-dev
17-
18-
RUN wget https://repo.radeon.com/amdgpu-install/6.1/ubuntu/jammy/amdgpu-install_6.1.60100-1_all.deb && \
19-
apt install -yqq ./amdgpu-install_6.1.60100-1_all.deb && \
20-
amdgpu-install -y --usecase=rocmdev && \
15+
# Install ROCM
16+
17+
# Make the directory if it doesn't exist yet.
18+
# This location is recommended by the distribution maintainers.
19+
RUN mkdir --parents --mode=0755 /etc/apt/keyrings
20+
# Download the key, convert the signing-key to a full
21+
# keyring required by apt and store in the keyring directory
22+
RUN wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
23+
gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null && \
24+
# Add rocm repo
25+
echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/6.1.1 jammy main" \
26+
| tee --append /etc/apt/sources.list.d/rocm.list && \
27+
printf 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | tee /etc/apt/preferences.d/rocm-pin-600 && \
28+
apt update
29+
# Install the kernel driver
30+
RUN apt install -yqq rocm-dev && \
2131
apt-get clean && \
2232
rm -rf /var/lib/apt/lists/*
2333

0 commit comments

Comments
 (0)