XGBoost is an optimized distributed gradient boosting library designed to be highly efficient, flexible and portable. It is a popular package used for a wide variety of machine learning and datascience tasks, serving the role of a convenient, domain-agnostic black box classifier. XGBoost provides GPU accelerated learning for some problems, and Compute Canada provides a GPU enabled build.
Python Module Installation
A very common way to use XGBoost is though its python interface, provided as the
xgboost python module. Compute Canada provides an optimized, multi-GPU enabled build as a Python wheel; readers can should familiarize themselves with the use of Python virtual environments before starting an XGBoost project.
Currently, version 0.81 of XGBoost is available. The following commands illustrate the needed package and module:
(myvenv) name@server $ module load nixpkgs/16.09 intel/2018.3 cuda/10.0.130 (myvenv) name@server $ module load nccl/2.3.5 (myvenv) name@server $ pip install xgboost==0.81 --no-index