Skip to content

Latest commit

 

History

History
80 lines (55 loc) · 2.08 KB

installation.md

File metadata and controls

80 lines (55 loc) · 2.08 KB
Raw

Installation

Table of Contents

Installation with pip

To install TensorFlow Decision Forests, run:

pip3 install tensorflow_decision_forests --upgrade

Then, check the installation with:

python3 -c "import tensorflow_decision_forests as tfdf; print('Found TF-DF v' + tfdf.__version__)"

Note: Cuda warnings are not an issue.

Build from source

Requirements

  • Microsoft Visual Studio >= 2019 (Windows) or GCC>=7.3 (Linux).
  • Bazel >= 3.7.2
  • Python >= 3
  • Git
  • Python's numpy
  • MSYS2 (Windows only)

Download and compile TensorFlow Decision Forests as follow:

# Download the source code of TF-DF.
git clone https://github.com/tensorflow/decision-forests.git
cd decision-forests

# Download TensorFlow's BazelRC.
# See "tensorflow_bazelrc" for the compilation options.
# Note: If you compile with an older version of TensorFlow, download the
# corresponding bazelrc.
TENSORFLOW_BAZELRC="tensorflow_bazelrc"
wget https://raw.githubusercontent.com/tensorflow/tensorflow/master/.bazelrc -O ${TENSORFLOW_BAZELRC}

# Compile TensorFlow Decision Forests.
bazel-3.7.2 --bazelrc=${TENSORFLOW_BAZELRC} build //tensorflow_decision_forests/...:all --config=linux

# Create a pip package.
./tools/build_pip_package.sh simple

# Install the pip package
pip3 install dist/*.whl

Note: Compiling TensorFlow Decision Forests compiles a large part of TensorFlow. This operation will take multiple hours without distributed building. See "tensorflow_bazelrc" for the distributed compilation instructions.

Troubleshooting

Note: Check also Yggdrasil's Troubleshooting page.