Windows安装xgboost gpu版

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/juwikuang/article/details/102635824

准备

首先，安装cuda。我这里装的是10.0版，因为tensorflow现在要求10.0版。建议您安装xgboost gpu版的时候，注意看一下tensorflow的安装文档

cuda

Visual Studio 2017, 这个比较麻烦，我是在自己电脑上找到installer的。。。

git for windows

cmake

下载源代码

如果你在国外，请用这个命令下载

git clone https://github.com/dmlc/xgboost.git xgboost

按照中国这个破网速，还是下载zip比较快
https://github.com/dmlc/xgboost/archive/master.zip

git submodule init
git submodule update

编译

编译之前，看一下自己的c++版本号，cuda版本号。

在源代码文件夹，敲以下命令。根据您的情况修改版本号。

mkdir build
cd build
"C:\Program Files\CMake\bin\cmake.exe" ..  -G"Visual Studio 15 2017 Win64" -T v141,cuda=10.0  -DUSE_CUDA=ON

编译成功

进入build文件夹，打开vs solution文件xgboost.sln

选择release 和 x64（我的windows 10是64位）

再次右键点击solution，点击第一个菜单条目Build Solution。等了几分钟，也许是十分钟，才编译好。

编译成功的dll位于xgboost文件夹的子文件夹lib下面

把他复制到另一个子文件夹python-package下面。

在python-package文件夹下面打

python setup.py install

测试

python -c "import xgboost"

参考文献

官方文档
stackoverflow