我想在Ubuntu上安装GPU版本的lightgbm,基于以下命令:
pip install lightgbm --install-option=--gpu
在安装过程中,出现错误,说"请先安装CMake"。安装CMake后,我再次收到同样的错误。为了确保安装了CMake,我运行以下命令并获得已安装的CMake的正确版本:
/opt/cmake/bin/cmake -version
出现此错误的可能原因是什么?
答案 0 :(得分:3)
我在安装了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
之后我能够成功运行Python和import lightgbm
。
“请安装CMake”可以掩盖其他错误。要查看有关错误的详细信息,请在LightGBM/python-package/setup.py
中的函数silent_call
中更改行
subprocess.check_output(cmd, stderr=shut_up)
到
subprocess.check_output(cmd)
并运行
python setup.py install --gpu
答案 1 :(得分:0)
我在点子7.1.0上遇到了这个错误。
pip 9.0.1的更新解决了这个问题!