Skip to content

Commit

Permalink
[train/release] Fix horovod user test master app config (ray-project#…
Browse files Browse the repository at this point in the history
krfricke authored May 15, 2022

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent 9381dd1 commit de69b0d
Showing 2 changed files with 19 additions and 1 deletion.
18 changes: 18 additions & 0 deletions release/ml_user_tests/horovod/app_config_master.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
base_image: "anyscale/ray-ml:nightly-py37-gpu"
env_vars: {}
debian_packages:
- curl

python:
pip_packages:
- pytest
- awscli
conda_packages: []

post_build_cmds:
- pip3 uninstall ray -y || true
- pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
- pip3 install 'ray[tune]'
- pip3 install torch torchvision
- HOROVOD_WITH_GLOO=1 HOROVOD_WITHOUT_MPI=1 HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITH_PYTORCH=1 pip3 install -U git+https://github.com/horovod/horovod.git
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
2 changes: 1 addition & 1 deletion release/release_tests.yaml
Original file line number Diff line number Diff line change
@@ -593,7 +593,7 @@
team: ml

cluster:
cluster_env: ../horovod_tests/app_config_master.yaml
cluster_env: horovod/app_config_master.yaml
cluster_compute: horovod/compute_tpl.yaml

driver_setup: horovod/driver_setup_master.sh

0 comments on commit de69b0d

Please sign in to comment.