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