Download the Software
The RAPIDS team is developing GPU enhancements to open-source XGBoost, working closely with the DCML/XGBoost organization to improve the larger ecosystem. Since RAPIDS is iterating ahead of upstream XGBoost releases, some enhancements will be available earlier from the RAPIDS branch, or from RAPIDS-provided installers.
Installation Prerequisites for RAPIDS + XGBoost
CUDA-compatible NVIDIA GPU with the NVIDIA Pascal™, NVIDIA Volta, or NVIDIA Turing™ architecture
NVIDIA CUDA Toolkit version CUDA 9.2 with driver v396.37+ or CUDA 10.0 with driver v410.48+
The latest RAPIDS package, which can be downloaded and installed one of these ways: