使用R语言训练XGBoost模型时,为提高效率,打算运用GPU加速。我使用的是Win10系统,习惯用R语言,但这方面的中文教程比较少。我打算用这篇博文把摸索的过程和心得记录下来。
0 预备工作
- 这里参考了官方的教程:https://xgboost.readthedocs.io/en/latest/build.html#obtaining-the-source-code
- 首先要有支持CUDA 10.1及以上的GPU(我省吃俭用买了张3060ti),并且下载安装CUDA,安装教程可以参考其他的博文:https://blog.youkuaiyun.com/weixin_43848614/article/details/117221384
- 注意,在安装CUDA之前要安装Visual Studio
- 要下载并安装Git: https://git-scm.com/
- 提前安装Rtools,并且把下面的Rtools位置写进环境变量PATH中:
C:\rtools40\usr\bin
C:\rtools40\mingw64\bin - 下载并安装cmake:
https://cmake.org/download/

本文详细记录了在Windows 10系统中,使用R语言配合GPU加速训练XGBoost和LightGBM模型的过程。首先介绍了预备工作,包括安装CUDA、Visual Studio、Git、Rtools和cmake。接着详细讲解了XGBoost的下载与安装,以及如何配置环境以利用GPU。在模型训练中,通过设置参数实现GPU加速。同时,文章还提及了遇到的问题和解决方案,如CUDA工具集缺失、dlltools.exe找不到等问题,并提到了LightGBM的GPU版本安装。最后,作者提醒在调参时要注意显存管理和优化策略。
最低0.47元/天 解锁文章
2738

被折叠的 条评论
为什么被折叠?



