experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #46193 from Intel-tensorflow:ganand1/build_openssh_path_fix

Browse Source 
PiperOrigin-RevId: 350375677
Change-Id: I18a550699191428fdec146207a097d86d7cb8cc3
This commit is contained in:
TensorFlower Gardener 2021-01-06 10:07:45 -08:00
commit 9b3f838c23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/tools/ci_build/linux/mkl/install_openmpi_horovod.sh
View File

@ -25,7 +25,7 @@ OPENMPI_DOWNLOAD_URL=${OPENMPI_DOWNLOAD_URL:-https://www.open-mpi.org/software/o
INSTALL_HOROVOD_FROM_COMMIT=${INSTALL_HOROVOD_FROM_COMMIT:-no}
BUILD_SSH=${BUILD_SSH:-no}
HOROVOD_VERSION=${HOROVOD_VERSION:-0.19.1}
SSH_CONFIG_PATH=/etc/ssh/ssh_config
SSH_CONFIG_PATH=/etc/ssh
# Install Open MPI
echo "Installing OpenMPI version ${OPENMPI_VERSION} ..."
@ -71,7 +71,7 @@ if [[ ${BUILD_SSH} == "yes" ]]; then
apt-get clean && apt-get update && \
apt-get install -y --no-install-recommends --fix-missing \
libnuma-dev cmake
SSH_CONFIG_PATH=/usr/local/etc/ssh_config
SSH_CONFIG_PATH=/usr/local/etc
else
apt-get clean && apt-get update && \
apt-get install -y --no-install-recommends --fix-missing \
@ -90,10 +90,10 @@ else
fi
fi
mkdir -p /var/run/sshd
grep -v StrictHostKeyChecking ${SSH_CONFIG_PATH} > /etc/ssh/ssh_config.new
grep -v StrictHostKeyChecking ${SSH_CONFIG_PATH}/ssh_config > ${SSH_CONFIG_PATH}/ssh_config.new
# Allow OpenSSH to talk to containers without asking for confirmation
echo " StrictHostKeyChecking no" >> /etc/ssh/ssh_config.new
mv /etc/ssh/ssh_config.new ${SSH_CONFIG_PATH}
echo " StrictHostKeyChecking no" >> ${SSH_CONFIG_PATH}/ssh_config.new
mv ${SSH_CONFIG_PATH}/ssh_config.new ${SSH_CONFIG_PATH}/ssh_config
# Install Horovod
if [[ ${INSTALL_HOROVOD_FROM_COMMIT} == "yes" ]]; then