LightGBM的GPU版本安装

1. 安装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"

指定device=gpu后可以使用GPU

2. 安装CUDA版本

export path=/usr/local/cuda/bin:$PATH
pip install lightgbm --install-option=--cuda

指定device=cuda报错，提示需要cmake

python版本的LightGBM安装
LightGBM官方文档