使用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.csdn.net/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://