环境: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#