如何在Apple M1 Pro芯片上启用XGBoost的GPU加速

1. 安装支持Metal的XGBoost版本 #

XGBoost从1.6.0版本开始支持Apple Silicon芯片的Metal GPU加速，需确保安装对应版本：

• 通过conda安装：

  conda install -c conda-forge xgboost
  

• 通过pip安装：

  pip install xgboost
  

安装后可通过print(xgb.__version__)确认版本是否≥1.6.0。

2. 配置训练参数启用GPU #

无需设置cuda，直接将device参数指定为gpu即可调用M1 Pro的Metal GPU：

import xgboost as xgb

# 构造训练数据矩阵
dtrain = xgb.DMatrix(X_train, label=y_train)

# 定义训练参数，核心是指定device为gpu
train_params = {
    'objective': 'binary:logistic',  # 根据你的任务调整目标函数
    'device': 'gpu',
    'max_depth': 6,
    'learning_rate': 0.1,
    'eval_metric': 'logloss'
}

# 启动训练
model = xgb.train(train_params, dtrain, num_boost_round=100)

3. 关键注意事项 #

• 确保Python环境为原生Apple Silicon版本（非Rosetta模拟），否则无法正常调用Metal GPU
• 若自动启用失败，可手动设置环境变量强制开启Metal：

  export XGBOOST_USE_METAL=1
  

内容的提问来源于stack exchange，提问作者Rishabh K Sharma