很早就知道XGBoost支持GPU了,不过一直没有配置,一方面是GPU版本的xgb配置,中文安装教程根本找不到,还有就是支持GPU的xgb还不是稳定版的。
花了两天时间踩遍各种坑,特意整理好这个教程,方便各位。
安装CUDA
这应该算是第一个坑,我的电脑在装tensorflow的时候就已经配置好CUDA了,看了下官方文档
Windows requirements for GPU build: only Visual C++ 2015 or 2013 with CUDA v8.0 were fully tested. Either install Visual C++ 2015 Build Tools separately, or as a part of Visual Studio 2015.
刚好我电脑上已经装好了VS2015还有CUDA v8.0,结果在下面的构建过程中炸了,各种尝试后,装了最新版的CUDA v9.0,再也没有出过问题。
构建XGBoost
先说下官网的教程
官网教程链接 http://xgboost.readthedocs.io/en/latest/build.html
github上clone源码
git clone --recursive https://github.com/dmlc/xgboost
在根目录下新建文件夹build,并进入该文件夹执行cmake相关命令
mkdir build
cd build
cmake .. -G"Visual Studio 14 2015 Win64" -DUSE_CUDA=ON
然后cmake构建
cmake --build .