当前位置:   article > 正文

XGBoost GPU加速环境配置

xgboost gpu

目录

前言

二、配置步骤

1.使用CMake编译xgboost源码

2.测试

总结


前言

记录下XGBoost算法通过GPU加速的过程中遇到的问题


一、准备工作

首先我们需要知道XGBoost  GPU加速所需要的硬件设备:通过查阅文档:

XGBoost GPU Support — xgboost 2.0.0-dev documentation  

二、配置步骤

1.使用CMake编译xgboost源码

按照一下博客进行编译dll:

win10 XGBoost GPU加速_potun7890的博客-CSDN博客

重点记录下遇到的问题:

  • "could not find any instance of Visual Studio":解决方法:通过重新安装vs2015
  • Could NOT find Doxygen (missing: DOXYGEN_EXECUTABLE): 百度搜索安装 

2.测试

按照上述博客 python setup.py install 后,在  xgboost\tests\benchmark  下进行测试,测试发现错误:cudaErrorNoKernelImageForDevice: no kernel image is available for execution on the device.

意思是没有足够的内核映像,内存不够,,我的电脑GTX730,算力3.5,,可能是这个原因,后续买到GTX1080ti试一下,再更新结果......


总结

1.cmake编译dll,出现很多问题

2.DLL编译好之后,XGBoost无法实现加速

遇到问题要耐心寻找问题可能的原因,不可急躁......

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/430493
推荐阅读
相关标签
  

闽ICP备14008679号