Windows 10 上Xgboost GPU配置填坑过程

本文档详细介绍了在Windows 10上配置Xgboost GPU的步骤,包括安装VS2017、CUDA 9.1.85.3和cuDNN 7.1.2,以及使用CMake编译Xgboost源码。提供了编译好的dll下载链接,同时提到了tensorflow 1.7与CUDA 9.1的兼容性问题,以及GPU相关参数设置和测试。


 主要过程:

1.    安装vs(我装的是2017)

2.    安装cuda(我装的是cuda9.1.85.3,cudnn7.1.2)

3.    安装cmake

4.   下载xgboost代码并编译

5.    安装

6.     xgboost gpu相关参数及demo

其他配置的比如:vs2015/2013,cuda 9.0/8.0等版本

 

编译好的dll已经放到网盘上了,如果你的版本(主要看cuda/cudnn,dll是基于这个编译的)跟我已一样,恭喜你,你可以直接下载dll安装,进入第五步

链接:https://pan.baidu.com/s/1vU5Lq-LdDnn13HoQcxdCIA 密码:odks

cuda及tensorflow安装文件:

链接:https://pan.baidu.com/s/1BNyMsg_EwAqg60OpsMBG-Q 密码:8iyq


第一步:vs

安装vs需要注意的是一些组件的选择,cmake会依赖一些windows sdk,推荐网址:

### 如何在Windows系统上安装和配置支持GPUXGBoost #### 准备工作 为了确保能够顺利安装并运行带有GPU支持的XGBoost,在开始之前需确认已正确设置了CUDA环境。这可以通过命令提示符中的`nvcc --version`来验证CUDA的存在与否;若有返回关于NVCC的信息说明已经成功安装了CUDA工具包,否则就需要按照官方指南完成其部署过程[^4]。 #### 安装CUDA Toolkit 对于尚未配备CUDA的情况,应当先下载适用于当前NVIDIA显卡驱动程序版本的CUDA Toolkit。通过执行`nvidia-smi`获取设备详情以及兼容性的建议CUDA版本号之后,前往[NVIDIA官方网站](https://developer.nvidia.com/cuda-downloads),依据操作系统和个人硬件条件挑选合适的发行版进行安装。 #### 配置开发环境变量 安装完成后,还需要适当调整系统的PATH环境变量以便于访问新加入的库文件路径。具体来说就是把CUDA bin目录添加进去,比如典型的设置会涉及到如下几个位置: - `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXX.X\bin` - `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXX.X\libnvvp` 其中vXX.X代表具体的CUDA版本号,请根据实际情况替换相应部分。 #### 使用pip安装XGBoost-GPU 当上述准备工作全部就绪后,就可以利用Python包管理器pip来进行最终一步——即实际安装XGBoost本身了。考虑到性能优化的需求,推荐采用预编译好的二进制轮子而不是源码构建方式: ```bash pip install xgboost --upgrade --pre -f https://anaconda.org/rapidsai-nightly/xgboost/files ``` 这条命令将会自动拉取由RAPIDS团队维护的支持最新特性和改进措施的夜间构建版本,并且默认包含了必要的依赖项以启用GPU加速特性[^1]。 #### 测试安装效果 最后但同样重要的是要检验整个流程是否无误地完成了预期目标。创建一个新的Python脚本或交互式解释器会话,尝试导入xgboost模块并且初始化一个DMatrix对象时指定参数`tree_method='gpu_hist'`,以此观察是否存在任何异常报错现象或是警告消息。正常情况下应该没有任何阻碍地加载完毕而不会抛出错误例外事件[^2]。 ```python import xgboost as xgb data = [[1, 2], [3, 4]] label = ['A', 'B'] dtrain = xgb.DMatrix(data=data, label=label) params = {'objective': 'binary:logistic'} bst = xgb.train(params=params, dtrain=dtrain, num_boost_round=10, tree_method='gpu_hist') ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值