mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 01:10:22 +00:00
e7d5e652df
Co-authored-by: Olli Saarikivi <olsaarik@microsoft.com> Co-authored-by: Changho Hwang <changhohwang@microsoft.com> Co-authored-by: Binyang Li <binyli@microsoft.com>
37 lines
1.2 KiB
Docker
37 lines
1.2 KiB
Docker
FROM ghcr.io/microsoft/mscclpp/mscclpp:base-cuda12.1
|
|
|
|
LABEL maintainer="MSCCL++"
|
|
LABEL org.opencontainers.image.source https://github.com/microsoft/mscclpp
|
|
|
|
ENV MSCCLPP_HOME="/usr/local/mscclpp" \
|
|
MSCCLPP_SRC_DIR="/tmp/mscclpp" \
|
|
CMAKE_VERSION="3.26.4"
|
|
|
|
# Download cmake 3.26.4
|
|
ENV CMAKE_HOME="/tmp/cmake-${CMAKE_VERSION}-linux-x86_64" \
|
|
CMAKE_URL="https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-x86_64.tar.gz"
|
|
RUN curl -L ${CMAKE_URL} -o ${CMAKE_HOME}.tar.gz && \
|
|
tar xzf ${CMAKE_HOME}.tar.gz -C /tmp
|
|
|
|
# Install MSCCL++
|
|
ADD . ${MSCCLPP_SRC_DIR}
|
|
WORKDIR ${MSCCLPP_SRC_DIR}
|
|
RUN rm -rf build && \
|
|
mkdir build && \
|
|
cd build && \
|
|
${CMAKE_HOME}/bin/cmake -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=${MSCCLPP_HOME} .. && \
|
|
make -j mscclpp mscclpp_static && \
|
|
make install/fast && \
|
|
strip ${MSCCLPP_HOME}/lib/libmscclpp.so.[0-9]*.[0-9]*.[0-9]*
|
|
|
|
# Install MSCCL++ Python bindings
|
|
WORKDIR ${MSCCLPP_SRC_DIR}
|
|
RUN python3.8 -m pip install .
|
|
|
|
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${MSCCLPP_HOME}/lib"
|
|
RUN echo LD_LIBRARY_PATH="${LD_LIBRARY_PATH}" >> /etc/environment
|
|
|
|
# Cleanup
|
|
WORKDIR /
|
|
RUN rm -rf ${CMAKE_HOME}* ${MSCCLPP_SRC_DIR}
|