小编Ант*_*кий的帖子

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

我正在尝试在 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
Run Code Online (Sandbox Code Playgroud)


然后我遵循了该指南: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
Run Code Online (Sandbox Code Playgroud)


或者告诉我的 boost 和 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= …
Run Code Online (Sandbox Code Playgroud)

python boost gpu cmake lightgbm

2
推荐指数
1
解决办法
2426
查看次数

标签 统计

boost ×1

cmake ×1

gpu ×1

lightgbm ×1

python ×1