我想在 Ubuntu 上安装 GPU 版本的 lightgbm,基于以下命令:
pip install lightgbm --install-option=--gpu
在安装过程中,出现“请先安装 CMake”的错误。安装 CMake 后,我再次收到相同的错误。为确保已安装 CMake,我运行以下命令并获取已安装 CMake 的正确版本:
/opt/cmake/bin/cmake -version
这个错误的可能原因是什么?
在安装了 CUDA 和 cmake 版本 3.5.1 的 Ubuntu 16.04 上,我得到了这个错误。就我而言,尽管出现“请安装 CMake”错误,但问题是:
通过执行以下操作,我能够为 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
在我的 M1 MacBook Air 上,我也遇到了同样的问题。我通过使用来解决这个问题brew
。首先,我brew install gcc
后来brew install CMake
因为依赖而跑了。之后,我brew install lightgbm
在 IDE 的终端上运行。最后,pip install lightgbm
在我的 IDE 终端上运行命令,我可以成功导入 lightgbm。
我在 pip 7.1.0 上有这个错误。
pip 9.0.1 的更新解决了这个问题!