写篇文章以防后面装系统再忘了,浪费时间。
1、首先按照http://lightgbm.apachecn.org/cn/latest/Installation-Guide.html中使用命令行编译为exe和dll文件。
命令行未编译GPU,所以修改后的命令如下:
git clone --recursive https://github.com/Microsoft/LightGBM
cd LightGBM
mkdir build
cd build
cmake -DCMAKE_GENERATOR_PLATFORM=x64 -DUSE_GPU=1 ..
cmake --build . --target ALL_BUILD --config Release
2、在命令提示符下,到LightGBM\python-package目录下执行
pip install lightgbm --install-option=--gpu
即可。