mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 01:10:22 +00:00
20 lines
492 B
Docker
20 lines
492 B
Docker
ARG BASE_IMAGE
|
|
FROM ${BASE_IMAGE}
|
|
|
|
LABEL maintainer="MSCCL++"
|
|
LABEL org.opencontainers.image.source=https://github.com/microsoft/mscclpp
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
ENV RCCL_VERSION=rocm-6.2.0
|
|
ARG GPU_ARCH=gfx942
|
|
ENV ARCH_TARGET=${GPU_ARCH}
|
|
RUN cd /tmp && \
|
|
git clone --branch ${RCCL_VERSION} --depth 1 https://github.com/ROCm/rccl.git && \
|
|
cd rccl && \
|
|
./install.sh --prefix=/opt/rocm --amdgpu_targets ${ARCH_TARGET} && \
|
|
cd .. && \
|
|
rm -rf /tmp/rccl
|
|
|
|
WORKDIR /
|