ray/release/ml_user_tests/horovod/driver_setup_latest.sh
Amog Kamsetty 3408b60d2b
[Release] Refactor User Tests (#20028)
* wip

* add directory

* wip

* try again

* Revert "try again"

This reverts commit 82d33ccea6f92848df025e019b87df73cea49e5d.

* finish

* formatting

* fix merge

* fix path

* chmod

* check

* sudo

* wip

* update

* fix horovod

* try

* typo

* reduce num workers
2021-11-05 17:28:37 -07:00

10 lines
No EOL
267 B
Bash
Executable file

#!/bin/bash
sudo apt update
sudo apt -y install build-essential
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 horovod