Skip to content

Commit

Permalink
Merge branch 'ashahba/duplicate-block' into 'develop-ng'
Browse files Browse the repository at this point in the history
Remove duplicate mpi installation block

See merge request intelai/models!142
  • Loading branch information
ashahba committed May 5, 2020
2 parents 8371247 + 14cc991 commit 02a14e3
Showing 1 changed file with 2 additions and 30 deletions.
32 changes: 2 additions & 30 deletions benchmarks/common/tensorflow/start.sh
Original file line number Diff line number Diff line change
Expand Up @@ -77,40 +77,12 @@ if [[ ${NOINSTALL} != "True" ]]; then
fi

if [[ ${MPI_NUM_PROCESSES} != "None" ]]; then
## Installing OpenMPI
apt-get install openmpi-bin openmpi-common openssh-client openssh-server libopenmpi-dev -y
# Horovod Installation
export HOROVOD_WITHOUT_PYTORCH=1
export HOROVOD_WITHOUT_MXNET=1
# TODO: lock a horovod commit
pip install --no-cache-dir horovod
fi

if [[ ${MPI_NUM_PROCESSES} != "None" ]]; then
### Manually Installing OpenMPI3.0 In Case apt-get install fails
##mkdir /tmp/openmpi && \
##pushd /tmp/openmpi && \
##wget https://www.open-mpi.org/software/ompi/v3.0/downloads/openmpi-3.0.0.tar.gz && \
##tar zxf openmpi-3.0.0.tar.gz && \
##cd openmpi-3.0.0 && \
##./configure --enable-orterun-prefix-by-default && \
##make -j $(nproc) all && \
##make install && \
##ldconfig
##popd
##rm -rf /tmp/openmpi
### Install g++-4.8 for horovod 0.18.1
##apt install -y g++-4.8
### Horovod Installation
##export HOROVOD_WITHOUT_PYTORCH=1
##export HOROVOD_WITHOUT_MXNET=1
### lock to a horovod commit in 0.18.2
##pip install --no-cache-dir git+https://github.com/horovod/horovod@d40169415ca92c6a923003dbb27e508ba4202426
## Installing OpenMPI
# Installing OpenMPI
apt-get install openmpi-bin openmpi-common openssh-client openssh-server libopenmpi-dev -y
# Horovod Installation
export HOROVOD_WITHOUT_PYTORCH=1
export HOROVOD_WITHOUT_MXNET=1
export HOROVOD_WITH_TENSORFLOW=1
# lock horovod==0.19.1 release commit/version
pip install --no-cache-dir horovod==0.19.1
fi
Expand Down

0 comments on commit 02a14e3

Please sign in to comment.