5

我想在 Ubuntu 上安装 GPU 版本的 lightgbm,基于以下命令:

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

在安装过程中,出现“请先安装 CMake”的错误。安装 CMake 后,我再次收到相同的错误。为确保已安装 CMake,我运行以下命令并获取已安装 CMake 的正确版本:

/opt/cmake/bin/cmake -version

这个错误的可能原因是什么?

4

3 回答 3

5

在安装了 CUDA 和 cmake 版本 3.5.1 的 Ubuntu 16.04 上,我得到了这个错误。就我而言,尽管出现“请安装 CMake”错误,但问题是:

  • 未安装必要的 boost 库,并且
  • cmake 无法找到 OpenCL。

通过执行以下操作,我能够为 Python 安装 LightGBM:

# Install boost libraries.
sudo apt-get install libboost-all-dev
# Get LightGBM source.
git clone --recursive https://github.com/Microsoft/LightGBM.git
cd LightGBM/python-package/
# cmake specifying locations of OpenCL files.
sudo cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda-8.0/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda-8.0/include/ ..
# Compile.
sudo make
# Install for Python, using what we just compiled.
python setup.py install --precompile

之后,我能够import lightgbm成功运行 Python。

没有帮助?

“请安装 CMake”可以掩盖其他错误。要查看有关问题的更多详细信息,请LightGBM/python-package/setup.py在函数silent_call中更改行

subprocess.check_output(cmd, stderr=shut_up)

subprocess.check_output(cmd) 

并运行

python setup.py install --gpu
于 2017-11-18T15:26:49.480 回答
0

在我的 M1 MacBook Air 上,我也遇到了同样的问题。我通过使用来解决这个问题brew。首先,我brew install gcc后来brew install CMake因为依赖而跑了。之后,我brew install lightgbm在 IDE 的终端上运行。最后,pip install lightgbm在我的 IDE 终端上运行命令,我可以成功导入 lightgbm。

于 2022-02-12T21:24:17.957 回答
0

我在 pip 7.1.0 上有这个错误。

pip 9.0.1 的更新解决了这个问题!

于 2018-02-07T11:08:44.087 回答