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

  1. 安装CUDA版本
export path=/usr/local/cuda/bin:$PATH
pip install lightgbm --install-option=--cuda

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

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

你可能感兴趣的:(LightGBM的GPU版本安装)