Linux下安裝LightGBM-GPU版本

安裝依賴

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 paths to OpenCL headers and library like the following:
# cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ ..
make -j4

本人使用方法二成功安裝。
但是需要注意的是，還需要在你想要的環境裏進行這個步驟：

cd ../python-package/
python setup.py install