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

Add horovod tests

Browse Source 
PiperOrigin-RevId: 319145206
Change-Id: I44f186249e609caa570eef7d48b53da83214eb44
This commit is contained in:
A. Unique TensorFlower 2020-06-30 19:18:06 -07:00 committed by TensorFlower Gardener
parent 1bf8f49335
commit 3be438aca2
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tensorflow/tools/ci_build/horovod/gpu/nightly.sh
View File

@ -35,12 +35,32 @@ fi
wget https://download.open-mpi.org/release/open-mpi/v4.0/openmpi-4.0.4.tar.gz wget https://download.open-mpi.org/release/open-mpi/v4.0/openmpi-4.0.4.tar.gz
tar xvf openmpi-4.0.4.tar.gz tar xvf openmpi-4.0.4.tar.gz
cd openmpi # Install gcc.
sudo apt install --assume-yes build-essential
gcc --version
cd openmpi-4.0.4
./configure ./configure
# Install open-mpi. # Install open-mpi.
sudo make all install sudo make all install
export LD_LIBRARY_PATH=/usr/local/lib/openmpi
sudo ldconfig sudo ldconfig
# Install Horovod.
cd ..
pip3 install horovod tensorflow
# Install tests.
git clone https://github.com/DEKHTIARJonathan/TF_HVD_Stability_Test.git
# Install pytest.
pip3 install -U pytest
# Install requirements.
cd TF_HVD_Stability_Test
pip3 install -r requirements.txt
# Run the tests.
python3 -m pytest