安装依赖
sudo apt-get install --no-install-recommends git cmake build-essential libboost-dev libboost-system-dev libboost-filesystem-dev
安装库
pip install setuptools wheel numpy scipy scikit-learn -U
安装GPU版本的LightGBM
方法一
sudo pip3 install lightgbm --install-option=--gpu --install-option="--opencl-include-dir=/usr/local/cuda/include/" --install-option="--opencl-library=/usr/local/cuda/lib64/libOpenCL.so"
不知道大家怎么样,反正这个方法我一直不行。经过我后期分析,如果上述代码安装失败的化,估计使用sudo pip3 install lightgbm --install-option=--gpu
即可。
方法二
git clone --recursive https://github.com/microsoft/LightGBM ; cd LightGBM
mkdir build ; cd build
cmake -DUSE_GPU=1 ..
# if you have installed NVIDIA CUDA to a customized location, you should specify pa