diff --git a/release/horovod_tests/app_config_master.yaml b/release/horovod_tests/app_config_master.yaml index 2197c7d86..e5364acbf 100644 --- a/release/horovod_tests/app_config_master.yaml +++ b/release/horovod_tests/app_config_master.yaml @@ -13,5 +13,5 @@ post_build_cmds: - pip3 uninstall ray -y || true - pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }} - pip3 install 'ray[tune]' - - HOROVOD_WITH_GLOO=1 HOROVOD_WITHOUT_MPI=1 HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITH_PYTORCH=1 pip3 install -U horovod + - 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") }} diff --git a/release/ml_user_tests/horovod/driver_setup_master.sh b/release/ml_user_tests/horovod/driver_setup_master.sh index 6b8699eb6..98fae8712 100755 --- a/release/ml_user_tests/horovod/driver_setup_master.sh +++ b/release/ml_user_tests/horovod/driver_setup_master.sh @@ -6,4 +6,4 @@ pip install cmake pip install -U -r ./driver_requirements.txt -HOROVOD_WITH_GLOO=1 HOROVOD_WITHOUT_MPI=1 HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITH_PYTORCH=1 pip install git+https://github.com/horovod/horovod.git@06aa579c9966035453f92208706157dee14c14ab \ No newline at end of file +HOROVOD_WITH_GLOO=1 HOROVOD_WITHOUT_MPI=1 HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITH_PYTORCH=1 pip install -U git+https://github.com/horovod/horovod.git \ No newline at end of file