1 # Install Horovod 2 ARG HOROVOD_WITHOUT_PYTORCH=1 3 ARG HOROVOD_WITHOUT_MXNET=1 4 ARG HOROVOD_WITH_TENSORFLOW=1 5 ARG HOROVOD_VERSION=v0.21.1 6 7 RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \ 8 build-essential \ 9 cmake \ 10 g++-8 \ 11 gcc-8 \ 12 git \ 13 ${PYTHON}-dev 14 15 RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 100 --slave /usr/bin/g++ g++ /usr/bin/g++-9 && \ 16 update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-8 800 --slave /usr/bin/g++ g++ /usr/bin/g++-8 17 18 RUN ${PYTHON} -m pip install git+https://github.com/horovod/horovod.git@${HOROVOD_VERSION}