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版的一跑显存不够...打扰了,小的这就换电脑


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM