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 看一下版本

 

cpu跑xgboost太慢了,配好gpu版的一跑显存不够...打扰了,小的这就换电脑

posted @ 2019-09-14 17:29  猪突猛进!!!  阅读(2446)  评论(0编辑  收藏  举报