我正在尝试在 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)