我正在尝试在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,但是总是会遇到相同的错误。如果有人向我解释是什么问题,我将非常感激。
答案 0 :(得分:1)
要安装具有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 ,并且(在填写源代码在哪里和在何处生成二进制文件字段之后),您应该看到像这样的东西:
如果未选中 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
.......
就这些。