LightGBM 支持 GPU 加速训练,但并不支持 GPU 加速推理(即预测)。然而,LightGBM 支持多 GPU 进行训练,但目前不支持在推理过程中利用多 GPU。
使用 GPU 加速训练
LightGBM 可以利用 GPU 加速训练,以下是一个简单的示例,展示如何配置 LightGBM 使用 GPU 进行训练。
import lightgbm as lgb
from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
# 加载示例数据
data = load_boston()
X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, test_size=0.2, random_state=42)
# 创建数据集
train_data = lgb.Dataset(X_train, label=y_train)
test_data = lgb.Dataset(X_test, label=y_test)
# 设置参数
params = {
'boosting_type': 'gbdt