lightgbm
pip install lightgbm --install-option=–gpu
新增传入参数: ‘device’ : ‘gpu’, ‘gpu_platform_id’:0, ‘gpu_device_id’:0
xgboost:pypi.org download
pip install xgboost.whl
新增传入参数: ‘gpu_id’:0, ‘tree_method’:‘gpu_hist’
GPU加速你的xgboost、lightgbm
最新推荐文章于 2025-05-27 14:00:55 发布