ray/release/ml_user_tests/horovod/driver_setup_latest.sh

12 lines
No EOL
291 B
Bash
Executable file

#!/bin/bash
cd "${0%/*}" || exit 1
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