Xgboost GPU配置
阅读原文时间:2023年07月16日阅读:1

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

手机扫一扫

移动阅读更方便

阿里云服务器
腾讯云服务器
七牛云服务器

你可能感兴趣的文章