树模型LightGBM CatBoost XGBoost的gpu单机多卡支持
LightGBM读文档没看到明确说多卡的设置好像要用到clinfo?哪位用过可否告诉我一下==CatBoost原生支持多gpu,直接设置参数:params['task_type'] = 'GPU'params['devices'] = '1,2,3' XGBoost可支持多gpu,需使用dask,可参考:https://github.com/dmlc/xgboost/blob/master/demo/dask/gpu_training.py...
原创
2020-08-19 19:23:10 ·
1115 阅读 ·
0 评论