xgboost配置GPU

说明:博主的显卡为3090,驱动为470.94,CUDA版本为11.4,cmake 3.3.2,gcc 7.3.1
以上版本配置xgboost(2.0.0)GPU可以直接调用
直接安装即可 pip install xgboost==2.0.0

1、测试GPU代码

如果测试不通过,考虑2、3

import xgboost as xgb
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# 加载示例数据集
data = load_breast_cancer

你可能感兴趣的:(算法,xgboost,gpu)