一、安装
安装步骤:
下载源码,编译
lib_lightgbm.so
(对于windows
则是lib_lightgbm.dll
)git clone --recursive https://github.com/Microsoft/LightGBM
# 对于 windows 用户,需要执行:
# git submodule init
# git submodule update
cd LightGBM
mkdir build
cd build
cmake ..
make -j4
如果需要支持
GPU
,则执行cmake
时指定:cmake -DUSE_GPU=1 ..
# 如果你已经安装了 NVIDIA OpenGL,则使用:
# cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ ..
如果想要
protobuf
来保存和加载模型,则先安装protobuf c++
版本,然后使用:cmake -DUSE_PROTO=ON ..
当前版本不支持该功能
安装
python
支持- 其中
--precompile
指示:已经编译过,不需要再次编译
cd python-package
sudo python setup.py install --precompile
- 其中
直接
pip
安装:pip install lightgbm
pip install --no-binary :all: lightgbm #从源码编译安装
pip install lightgbm --install-option=--mpi #从源码编译安装 MPI 版本
pip install lightgbm --install-option=--gpu #从源码编译安装 GPU 版本
pip install lightgbm --install-option=--gpu --install-option="--opencl-include-dir=/usr/local/cuda/include/" --install-option="--opencl-library=/usr/local/cuda/lib64/libOpenCL.so" #从源码编译安装,指定配置
#可选的配置有:
# boost-root
# boost-dir
# boost-include-dir
# boost-librarydir
# opencl-include-dir
# opencl-library