Xgboost GPU配置

Posted z1141000271

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Xgboost GPU配置相关的知识,希望对你有一定的参考价值。

眼残cmake版本配错了搞了半天,简单记录一下,老规矩,参考一下官方的文档.

git clone --recursive https://github.com/dmlc/xgboost
cd xgboost
mkdir build
cd build
cmake .. -DUSE_CUDA=ON
make -j4

cd  ..
cd python-package
python setup.py install

整体的命令行如上所述,在cmake编译的时候会提示cmake版本要高于3.12,这里给一个cmake3.14的安装流程

> 首先在官网https://cmake.org/download/ 下载对应的包
> 解压之后cd到对应的文件夹
依次执行下述命令
sudo ./bootstrap
sudo make
sudo make install

最后cmake --version 看一下版本

 

以上是关于Xgboost GPU配置的主要内容,如果未能解决你的问题,请参考以下文章

Ubuntu18.04 安装xgboost GPU版配置

Ubuntu18.04 安装xgboost GPU版配置

R语言构建xgboost模型:使用GPU加速xgboost模型构建

ubuntu 18 安装xgboost GPU版本

使用 CPU 工作人员进行数据处理,并使用带有 dask 的 GPU 工作人员训练 xgboost

xgboost 中出现警告