Skip to content

Commit 55c9137

Browse files
committed
Revert base-level dockerfile to master version
1 parent 0ddf074 commit 55c9137

File tree

1 file changed

+2
-23
lines changed

1 file changed

+2
-23
lines changed

Dockerfile

Lines changed: 2 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,6 @@ FROM requirements AS requirements-drivers
1919
ARG BUILD_TYPE
2020
ARG CUDA_MAJOR_VERSION=12
2121
ARG CUDA_MINOR_VERSION=0
22-
ARG ROCM_MAJOR_VERSION=6
23-
ARG ROCM_MINOR_VERSION=1 # ROCm version to append to the major version, in the format of their apt repo (https://repo.radeon.com/rocm/apt/). Like `0_alpha` or `3.4`.
2422
ARG SKIP_DRIVERS=false
2523
ARG TARGETARCH
2624
ARG TARGETVARIANT
@@ -113,29 +111,10 @@ RUN if [ "${BUILD_TYPE}" = "clblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then \
113111
; fi
114112

115113
RUN if [ "${BUILD_TYPE}" = "hipblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then \
116-
# Setup for specific ROCm version as described here: https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/install-methods/package-manager/package-manager-ubuntu.html
117-
ROCM_VERSION="${ROCM_MAJOR_VERSION}.${ROCM_MINOR_VERSION}" && \
118114
apt-get update && \
119115
apt-get install -y --no-install-recommends \
120-
gpg wget && \
121-
mkdir --parents --mode=0755 /etc/apt/keyrings && \
122-
wget -qO - https://repo.radeon.com/rocm/rocm.gpg.key | gpg --yes --dearmor --output /etc/apt/keyrings/rocm.gpg && \
123-
echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/${ROCM_VERSION} jammy main" >> /etc/apt/sources.list.d/rocm.list && \
124-
if [ "${ROCM_MAJOR_VERSION}" -ge 7 ]; then \
125-
echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/graphics/${ROCM_VERSION}/ubuntu jammy main" >> /etc/apt/sources.list.d/rocm.list \
126-
; fi && \
127-
echo "Package: *" >> /etc/apt/preferences.d/rocm-pin-600 && \
128-
echo "Pin: release o=repo.radeon.com" >> /etc/apt/preferences.d/rocm-pin-600 && \
129-
echo "Pin-Priority: 600" >> /etc/apt/preferences.d/rocm-pin-600 && \
130-
# End setup steps for specific ROCm version
131-
apt-get update && \
132-
apt-get install -y --no-install-recommends \
133-
# Build dependencies
134-
rocm-developer-tools \
135-
rocm-hip-runtime-dev \
136-
rocm-hip-sdk \
137-
# Metapackage for the ROCm runtime + client tools
138-
rocm && \
116+
hipblas-dev \
117+
rocblas-dev && \
139118
apt-get clean && \
140119
rm -rf /var/lib/apt/lists/* && \
141120
echo "amd" > /run/localai/capability && \

0 commit comments

Comments
 (0)