我如何解决在Windows 10上安装lightgbm gpu的问题?

时间:2019-08-08 12:09:11

标签: python boost cmake gpu lightgbm

我正在尝试在Windows 10 pro x64上安装lightgbm gpu。我的GPU是NVIDIA Geforce GTX 950。

我尝试首先遵循本指南:https://lightgbm.readthedocs.io/en/latest/GPU-Windows.html 我用Mingw进行编译的地方。结果是我可以在python中成功导入LightGBM,但是当我尝试使用参数device ='gpu'拟合模型时,出现错误:

LightGBMError: GPU Tree Learner was not enabled in this build.
Please recompile with CMake option -DUSE_GPU=1


然后我按照那个指南: https://github.com/Microsoft/LightGBM/blob/master/docs/Installation-Guide.rst#build-gpu-version 在使用Visual Studio 2019成功编译后,我无法安装gpu版本的lightgbm:我总是会得到与此处完全相同的错误: https://github.com/Microsoft/LightGBM/issues/1121

我键入是否都没关系

pip install lightgbm --install-option=--gpu


或告诉我升压和CUDA的路径

pip install lightgbm --install-option=--gpu --install-option="--opencl- 
include-dir=C:/Program Files/NVIDIA GPU Computing 
Toolkit/CUDA/v10.1/include" --install-option="--opencl-library=C:/Program 
Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/lib/x64/OpenCL.lib" -- 
install-option="--boost-librarydir= C:/local/boost_1_69_0/lib64-msvc-14.1" 
--install-option="--boost-root= C:/local/boost_1_69_0"

我也尝试了各种版本的boost,但是总是会遇到相同的错误。如果有人向我解释是什么问题,我将非常感激。

1 个答案:

答案 0 :(得分:1)

在Python / Anaconda / Windows 8.1 / 10 x64上安装具有GPU支持的Lightgbm的终极指南。

要安装具有GPU支持的 Lightgmb ,您需要从源代码重建,并且没有其他解决方法。

您需要的东西:

1) Visual Studio 20xx xx > = 15,社区可以。)

在Windows 8.1上:需要另外安装 Windows SDK版本8.1 。它没有安装在标准安装中,因此您可以进行自定义安装,或者如果已经安装了自定义安装,则可以通过添加Windows SDK版本8.1来修改安装。

2)下载Booster文件(版本1_56_0或更高版本)并将其安装在默认文件夹C:/locals中。您需要在某些命令中更改这些数字 ,以分别匹配您自己的 Booster Visual Studio 版本。 当心

3)Cmake

4)Git

如果要在 Anaconda 上安装,请先激活正确的环境,然后再继续。

现在进入命令行:

git clone --recursive https://github.com/microsoft/LightGBM   
cd LightGBM
mkdir build
cd build
cmake -A x64 -DUSE_GPU=1 -DBOOST_ROOT= -DBOOST_LIBRARYDIR=C:/local/boost_1_71_0/lib64-msvc-14.0 .. 

-- Building for: Visual Studio 14 2015
-- Selecting Windows SDK version  to target Windows 10.0.18362.
-- The C compiler identification is MSVC 19.0.24215.1
   ...................
   ...................
-- Build files have been written to: J:/LightGBM/build

现在您调用 CMake GUI ,并且(在填写源代码在哪里在何处生成二进制文件字段之后),您应该看到像这样的东西:

enter image description here

如果未选中 USE_GPU USE_OPENMP ,请先进行检查。

点击配置,然后点击生成

关闭CMake。

现在输入

cmake --build . --target ALL_BUILD --config Release

这时,您会发现 LightGBM 文件夹中有一个 Release 文件夹。很好。

然后, 最终

cd ..\python-package
python setup.py install --gpu

running install
INFO:LightGBM:Starting to compile the library.
INFO:LightGBM:Starting to compile with Visual Studio 16 2019.
INFO:LightGBM:Starting to compile with Visual Studio 15 2017.
INFO:LightGBM:Starting to compile with Visual Studio 14 2015.
running build
running build_py
creating build
creating build\lib
creating build\lib\lightgbm
.......

就这些。