sh1ng / arboretum

Gradient Boosting powered by GPU(NVIDIA CUDA)

arboretum - Gradient Boosting implementation with focus on overcoming GRAM size limit

Installation wheel package

pip install arboretum

Benchmark results

Dependencies

• Python 2.7 or Python 3
• Cuda toolkit 7+
• Cuda cub https://github.com/NVlabs/cub as a submodule
• JSON for Modern C++ https://github.com/nlohmann/json as a submodule

Installation from source

• git clone --recursive https://github.com/sh1ng/arboretum.git
• $ mkdir build && cd build && cmake .. && make -j && cd .. && make wheel
• $ sudo python -m pip install python-package/dist/arboretum*.whl