📅  最后修改于: 2023-12-03 14:52:24.186000             🧑  作者: Mango
LightGBM 是一个开源的分布式梯度提升框架,具有高效、低内存占用和高精度等特点,广泛应用于机器学习、推荐系统等领域。本文将介绍如何在 Linux 上安装 LightGBM。
首先,需要安装一些必要的依赖:
sudo apt install cmake g++
sudo apt install libboost-dev libboost-system-dev libboost-filesystem-dev
在安装完依赖后,可以从 GitHub 上下载 LightGBM 的源代码:
git clone --recursive https://github.com/microsoft/LightGBM
这里需要注意,LightGBM 的源代码采用了子模块的方式管理,加上 --recursive
参数可以将子模块也一起下载。
下载完成后,进入源代码目录,执行以下命令进行编译:
cd LightGBM
mkdir build
cd build
cmake ..
make -j4
这里的 -j4
表示使用 4 个线程进行编译,可以根据自己的 CPU 核数进行调整。编译完成后,可以运行以下命令进行测试:
cd ../examples/cpp_classification
mkdir build
cd build
cmake ..
make -j4
./lightgbm
如果测试运行成功,就可以将编译好的库文件和头文件拷贝到系统库目录和头文件目录中:
sudo cp ../lib_lightgbm.so /usr/local/lib
sudo cp ../include/* /usr/local/include
除了 C++ 的接口外,LightGBM 也提供了 Python 包以供调用。可以使用 pip 命令安装:
pip install lightgbm
安装完成后,可以在 Python 中导入 LightGBM 库并进行使用:
import lightgbm as lgb
至此,LightGBM 在 Linux 上的安装就完成了。
本文介绍了如何在 Linux 上安装 LightGBM,主要包括以下几个步骤:
希望本文对大家有所帮助。