LightGBM gpu版本win10下安装

写篇文章以防后面装系统再忘了,浪费时间。

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

即可。

你可能感兴趣的:(LightGBM gpu版本win10下安装)