xgboost配置GPU_xgboost gpu

作者：weixin_40725706 | 2024-04-15 22:30:18

踩

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_cancer1
2
3
4
5
6

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/430507