一、安装

  1. 安装步骤:

    • 下载源码,编译lib_lightgbm.so (对于windows 则是 lib_lightgbm.dll

      1. git clone --recursive https://github.com/Microsoft/LightGBM
      2. # 对于 windows 用户,需要执行:
      3. # git submodule init
      4. # git submodule update
      5. cd LightGBM
      6. mkdir build
      7. cd build
      8. cmake ..
      9. make -j4

      如果需要支持GPU,则执行cmake 时指定:

      1. cmake -DUSE_GPU=1 ..
      2. # 如果你已经安装了 NVIDIA OpenGL,则使用:
      3. # cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ ..

      如果想要protobuf 来保存和加载模型,则先安装protobuf c++版本,然后使用:

      1. cmake -DUSE_PROTO=ON ..

      当前版本不支持该功能

    • 安装python 支持

      • 其中 --precompile 指示:已经编译过,不需要再次编译
      1. cd python-package
      2. sudo python setup.py install --precompile
  2. 直接pip 安装:

    1. pip install lightgbm
    2. pip install --no-binary :all: lightgbm #从源码编译安装
    3. pip install lightgbm --install-option=--mpi #从源码编译安装 MPI 版本
    4. pip install lightgbm --install-option=--gpu #从源码编译安装 GPU 版本
    5. 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" #从源码编译安装,指定配置
    6. #可选的配置有:
    7. # boost-root
    8. # boost-dir
    9. # boost-include-dir
    10. # boost-librarydir
    11. # opencl-include-dir
    12. # opencl-library