LightBGM和XgBoost是機器學習比賽的常客,二者在安裝上十分相似,此文將介紹GPU版本的LightBGM安裝。
本文環境:
Centos7.6 Python3.6.8 LightBGM2.3.2
1. 基礎環境
官網教程:https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html#id17
官網給出了安裝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的安裝也需要這一步。