Centos7安裝LightBGM(GPU版本)

LightBGM和XgBoost是機器學習比賽的常客,二者在安裝上十分相似,此文將介紹GPU版本的LightBGM安裝。

本文環境:

Centos7.6  Python3.6.8 LightBGM2.3.2

1. 基礎環境

官網教程:https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html#id17

yH5BAAAAAAALAAAAAAOAA4AAAIMhI+py+0Po5y02qsKADs=wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

官網給出了安裝GPU版本LightBGM,所需環境:

* OpenCL:

用過TF、PyTorch的童鞋肯定都很熟悉Cuda,Cuda自帶OpenCL;所以安裝Cuda即可;

* libboost

可以使用yum安裝,但是yum版本低,編譯會報錯;

官網下載boost庫,boost_1_71_0.tar.gz 鏈接:https://dl.bintray.com/boostorg/release/1.71.0/source/

下載並解壓完成後,進入文件夾,3步,先執行./bootstrap.sh ,再執行./b2,最後執行./b2 install;搞定;

* CMake 略

2. 安裝步驟

官網給出了安裝步驟:

但是,進入build文件夾後,會提示找不到CMakeLists.txt文件,所以,直接在LightBGM文件夾下編譯即可;

可以自己指定OpenCL安裝位置;

make -j4執行成功之後,需要將LightBGM安裝到Python3.6.8環境中;進入python-package文件夾,執行

python3.6.8 setup.py install,執行成功則安裝完成。xgboost的安裝也需要這一步。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章