/tf-collective-all-reduce

Primary LanguagePythonApache License 2.0Apache-2.0

Installation

python3.6 -m venv tf_env
. tf_env/bin/activate
pip install tensorflow==1.12.2
./install.sh
pip install -e .

ReBuild

Compiles sources and executes example below as healthcheck

./rebuild.sh

Run locally

pytest tests/test_simple_estimator.py