xgboost配置GPU加速(快速安装版)

环境:anaconda(python3.6) + win10

第一步:

下载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版安装成功。

GPU版使用方法:
param[‘gpu_id’] = 0
param[‘max_bin’] = 16
param[‘tree_method’] = ‘gpu_hist’

详情请查看xgboost API
http://xgboost.readthedocs.io/en/latest/gpu/index.html#

你可能感兴趣的:(机器学习)