xgboost进行GPU加速_xgboost gpu build

官方链接

https://github.com/dmlc/xgboost/tree/master/plugin/updater_gpu

第一步：下载CUB https://nvlabs.github.io/cub/ 

第二步：安装新版本cmake

查看cmake版本：

$ cmake --version

卸载旧版本：

$ sudo apt-get autoremove cmake

安装新版本cmake：

$ wget https://cmake.org/files/v3.5/cmake-3.5.2-Linux-x86_64.tar.gz

$ tar -xzvf cmake-3.5.2-Linux-x86_64.tar.gz

$ cmake --version

第三步：在xgboost根目录下运行以下代码

$ mkdir build

$ cd build

$ cmake .. -DPLUGIN_UPDATER_GPU=ON -DCUB_DIRECTORY=<MY_CUB_DIRECTORY>

<MY_CUB_DIRECTORY>指的是cub的文件路径，比如我的路径是/home/sys418/cub-1.6.4，那么第三句代码就是这样的：

cmake .. -DPLUGIN_UPDATER_GPU