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配置的主要内容,如果未能解决你的问题,请参考以下文章