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
GPU: NVIDIA Pascal™ or better with compute capability 6.0+
CUDA & NVIDIA Drivers: One of the following supported versions:
9.2 & v396.37+ 10.0 & v410.48+ 10.1.2 & v418.87+
The latest RAPIDS package, which can be downloaded and installed one of these ways: