Skip to content

Commit

Permalink
move horovod installation to multi-gpu
Browse files Browse the repository at this point in the history
  • Loading branch information
edknv committed Mar 7, 2023
1 parent dbb7af7 commit 53d0311
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ commands =
; Runs in: Github Actions
; Runs GPU-based tests.
allowlist_externals =
bash
horovodrun
deps =
-rrequirements/test.txt
Expand All @@ -41,12 +42,15 @@ commands =
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/core.git@{posargs:main}
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/dataloader.git@{posargs:main}
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/nvtabular.git@{posargs:main}
# Reinstall Horovod using `pip install --no-cache-dir` to build with the new version.
python -m pip install horovod --no-cache-dir
horovodrun --check-build
# Install distributed embeddings and check build
bash examples/usecases/multi-gpu/install_distributed_embeddings.sh {envtmpdir}
# Run multi-gpu tests marked with `horovod` marker
horovodrun -np 2 sh examples/usecases/multi-gpu/hvd_wrapper.sh pytest -m horovod -rxs tests/unit

[testenv:py38-horovod-cpu]
allowlist_externals =
bash
setenv =
HOROVOD_WITH_MPI=1
HOROVOD_WITH_TENSORFLOW=1
Expand All @@ -55,11 +59,6 @@ setenv =
commands =
conda update --yes --name base --channel defaults conda
conda env create --prefix {envdir}/env --file requirements/horovod-cpu-environment.yml --force
# Install horovod and check build
{envdir}/env/bin/python -m pip install horovod --no-cache-dir
{envdir}/env/bin/horovodrun --check-build
# Install distributed embeddings and check build
bash examples/usecases/multi-gpu/install_distributed_embeddings.sh {envtmpdir}
# Install Merlin packages
{envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/core.git
{envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/dataloader.git
Expand Down

0 comments on commit 53d0311

Please sign in to comment.