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的安装也需要这一步。