1.下载源码
git clone --recursive https://github.com/dmlc/xgboost
–recursive 递归clone,不加的话gputreeshape文件夹会为空,导致GPU版本编译失败
2.编译
2.1依赖工具版本
- cmake version 3.16.3
- GNU Make 4.2.1
2.2 编译
在xgboost目录
mkdir build
cd build
cmake .. -DUSE_CUDA=ON
make -j4
3.安装python接口
3.1 安装
在xgboost/python-package目录
python setup.py install