mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00

* Revert "Revert "Support of scikit-learn with ray joblib backend (#6925)" (#6957)"
This reverts commit 86100bc119
.
* adding scikit-learn to dependencies
63 lines
2.5 KiB
Bash
Executable file
63 lines
2.5 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
ROOT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
|
|
|
|
echo "PYTHON is $PYTHON"
|
|
|
|
platform="unknown"
|
|
unamestr="$(uname)"
|
|
if [[ "$unamestr" == "Linux" ]]; then
|
|
echo "Platform is linux."
|
|
platform="linux"
|
|
elif [[ "$unamestr" == "Darwin" ]]; then
|
|
echo "Platform is macosx."
|
|
platform="macosx"
|
|
else
|
|
echo "Unrecognized platform."
|
|
exit 1
|
|
fi
|
|
|
|
if [[ "$PYTHON" == "3.5" ]] && [[ "$platform" == "linux" ]]; then
|
|
sudo apt-get update
|
|
sudo apt-get install -y python-dev python-numpy build-essential curl unzip tmux gdb
|
|
# Install miniconda.
|
|
wget -q https://repo.continuum.io/miniconda/Miniconda3-4.5.4-Linux-x86_64.sh -O miniconda.sh -nv
|
|
bash miniconda.sh -b -p $HOME/miniconda
|
|
export PATH="$HOME/miniconda/bin:$PATH"
|
|
pip install -q scipy tensorflow cython==0.29.0 gym opencv-python-headless pyyaml pandas==0.24.2 requests \
|
|
feather-format lxml openpyxl xlrd py-spy setproctitle pytest-timeout networkx tabulate psutil aiohttp \
|
|
uvicorn dataclasses pygments werkzeug kubernetes flask grpcio pytest-sugar pytest-rerunfailures pytest-asyncio \
|
|
blist torch torchvision scikit-learn
|
|
elif [[ "$PYTHON" == "3.5" ]] && [[ "$platform" == "macosx" ]]; then
|
|
# Install miniconda.
|
|
wget -q https://repo.continuum.io/miniconda/Miniconda3-4.5.4-MacOSX-x86_64.sh -O miniconda.sh -nv
|
|
bash miniconda.sh -b -p $HOME/miniconda
|
|
export PATH="$HOME/miniconda/bin:$PATH"
|
|
pip install -q cython==0.29.0 tensorflow gym opencv-python-headless pyyaml pandas==0.24.2 requests \
|
|
feather-format lxml openpyxl xlrd py-spy setproctitle pytest-timeout networkx tabulate psutil aiohttp \
|
|
uvicorn dataclasses pygments werkzeug kubernetes flask grpcio pytest-sugar pytest-rerunfailures pytest-asyncio \
|
|
blist torch torchvision scikit-learn
|
|
elif [[ "$LINT" == "1" ]]; then
|
|
sudo apt-get update
|
|
sudo apt-get install -y build-essential curl unzip
|
|
# Install miniconda.
|
|
wget -q https://repo.continuum.io/miniconda/Miniconda3-4.5.4-Linux-x86_64.sh -O miniconda.sh -nv
|
|
bash miniconda.sh -b -p $HOME/miniconda
|
|
export PATH="$HOME/miniconda/bin:$PATH"
|
|
# Install Python linting tools.
|
|
pip install -q flake8==3.7.7 flake8-comprehensions flake8-quotes==2.0.0
|
|
elif [[ "$LINUX_WHEELS" == "1" ]]; then
|
|
sudo apt-get install docker
|
|
sudo usermod -a -G docker travis
|
|
elif [[ "$MAC_WHEELS" == "1" ]]; then
|
|
:
|
|
else
|
|
echo "Unrecognized environment."
|
|
exit 1
|
|
fi
|
|
|
|
if [[ "$PYTHON" == "3.5" ]] || [[ "$MAC_WHEELS" == "1" ]]; then
|
|
# Install the latest version of Node.js in order to build the dashboard.
|
|
source $HOME/.nvm/nvm.sh
|
|
nvm install node
|
|
fi
|