Xgboost实现GPU加速

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Einstellung/article/details/81638376

环境配置

首先最好你有帕斯卡构架的GPU,如果没有的话,最好还是不要搞GPU加速了。

首先安装CUDA

第二步:

下载xgboost源码:
https://github.com/dmlc/xgboost

第三步:

http://ssl.picnet.com.au/xgboost/中下载支持GPU版已编译好的DLL文件

第四步:

将xgboost.dll复制到xgboost-master/python-package/xgboost目录下,
打开控制台,进入到python-package目录下,执行python setup.py install命令,
至此xgboost GPU版安装成功。

需要注意的是,需要先把源码文件放到类似于这样的c:\users\keked\appdata\local\programs\python\python36\lib\site-packages路径里面。

猜你喜欢

转载自blog.csdn.net/Einstellung/article/details/81638376