Skip to content

Instantly share code, notes, and snippets.

@kingspp

kingspp/horovod_installation.sh

Last active January 12, 2019 20:22
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save kingspp/055345ec7f2fc61ea3e677621584f26d to your computer and use it in GitHub Desktop.
Save kingspp/055345ec7f2fc61ea3e677621584f26d to your computer and use it in GitHub Desktop.
Download ZIP
Horovod Installation with Open MPI Configuration
Raw
horovod_installation.sh
#!/usr/bin/env bash
# Install Open MPI v4
wget https://download.open-mpi.org/release/open-mpi/v4.0/openmpi-4.0.0.tar.gz
tar -xvf openmpi-4.0.0.tar.gz
cd openmpi-4.0.0
./configure --prefix=/usr/local
make all
sudo make install
sudo ldconfig
# Install Python Requirements
sudo apt-get install python3-dev
sudo pip3 install setuptools
# Install Tensorflow v1.8.0 GPU
sudo python3 -m pip3 install tensorflow-gpu==1.8.0
# Install Horovod
sudo python3 -m pip3 install horovod
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment