win10 XGBoost GPU加速

不念不忘少年蓝@ 2022-11-28 13:48 163阅读 0赞

[参考文献：https://zhuanlan.zhihu.com/p/29904369][https_zhuanlan.zhihu.com_p_29904369]

需要平台：win10,CUDA,VS2015,CMake

**目录**

1. 下载xgboost源码

2.安装cmake

3.生成构建文件

--------------------

### 1.下载xgboost源码  ###

可用git，没有git工具的直接去github上下载解压

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

### 2.安装cmake ###

[cmake.org/download/][cmake.org_download]

浏览器上下载巨慢，建议下载迅雷，然后右键‘复制链接地址’到迅雷，光速下载

[https://github.com/Kitware/CMake/releases/download/v3.18.1/cmake-3.18.1-win64-x64.msi][https_github.com_Kitware_CMake_releases_download_v3.18.1_cmake-3.18.1-win64-x64.msi]

### 3.生成构建文件 ###

*  打开桌面的**CMake (cmake-gui)**

分别选择源码目录以及源码目录下的build文件夹(没有就新建一个)

![20200818102143282.png][]

点击**Configure**，然后选择自己的VS版本，包括位数，点击**Finish**结束

**64位的记得选x64,不然后面编译cuda的时候会出错！！！**

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 1][]

把**USE\_CUDA**勾上，再点击**Configure **

如果64位会显示64

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 2][]

点击**Generate**，build文件夹下多了vs的工程文件

*  点击**Open Project**，会自动调用VS打开工程，直接右键生成解决方案等待就行

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 3][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 4][]

*  安装Python包

我用的anaconda，所以这里是直接进anaconda powershell中 cd到python-package目录

python setup.py install

~~~~~~~~~~~~~~~~~~花絮~~~~~~~~~~~~~~~~~~~~~

由于当时没选64位，默认成32位，出现bug:

No CUDA toolset found.

![20200818104058405.png][]

查看日志，一堆错误信息，包含：严重性 代码 说明 项目 文件 行 禁止显示状态 错误 MSB3721 ，和nvcc.exe，Microsoft Visual Studio等有关系的说明，还有“文件已退出，返回代码为 1”等

BTW:只有大型数据集使用XGBoost GPU版本才能加速，数据集小的说不定比CPU运行还慢

[https_zhuanlan.zhihu.com_p_29904369]: https://zhuanlan.zhihu.com/p/29904369
[cmake.org_download]: https://link.zhihu.com/?target=https%3A//cmake.org/download/
[https_github.com_Kitware_CMake_releases_download_v3.18.1_cmake-3.18.1-win64-x64.msi]: https://github.com/Kitware/CMake/releases/download/v3.18.1/cmake-3.18.1-win64-x64.msi
[20200818102143282.png]: /images/20221124/6aaa0b991f14463f858c70c166082222.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70]: /images/20221124/1a992127f43345b696a5e8202118700f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 1]: /images/20221124/08f57e6ce3e341488390f922eead0781.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 2]: /images/20221124/19576c93cee144d2b648b223925006d3.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 3]: /images/20221124/b5104886dd994e64b0e74d41a9cd0ee5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdHVuNzg5MA_size_16_color_FFFFFF_t_70 4]: /images/20221124/123dcf2c82414936a62e3c143aa3eff0.png
[20200818104058405.png]: /images/20221124/be114bfcbb2e4904ae32a14023ff6b29.png