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

发布了11 篇原创文章 · 获赞 3 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/xueqinmax/article/details/103481027