在 Google Collab 上为 LightGBM 安装 GPU 支持

Question

有人在尝试使用 Notebooks 在 Google Collab 上安装对 lightgbm 的 GPU 支持时很幸运吗？

Answer 1

很简单：运行即可

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

或者

pip install lightgbm --install-option=--gpu --install-option="--opencl-include-dir=/usr/local/cuda/include/" --install-option="--opencl-library=/usr/local/cuda/lib64/libOpenCL.so"

请记住在笔记本中启用 GPU 支持并添加'device':'gpu'lightgbm 设置。并且不要忘记先卸载不支持gpu版本的lightgbm版本。

Answer 2

确保您正确遵循了安装步骤

!git clone --recursive https://github.com/Microsoft/LightGBM
%cd LightGBM
!mkdir build
%cd build
!cmake ../../LightGBM
!make -j4

之后你必须执行 LightGBM 文件夹中的安装文件

%cd LightGBM/python-package
!python3 setup.py install --gpu

一旦完成，一切就都准备好了。ps：确保你已经安装了cmake，如果不是的话

!pip install cmake

Answer 3

其中大部分内容遵循此处提供的文档，并进行了两个小调整以使其在 Google Colab 上运行。
由于实例在使用 12 小时后会更新，因此我将其发布在笔记本的开头，以使用 lightgbm 重新安装 GPU 支持：

    !apt-get -qq install --no-install-recommends nvidia-375
    !apt-get -qq install --no-install-recommends nvidia-opencl-icd-375 nvidia-opencl-dev opencl-headers
    !apt-get -qq install --no-install-recommends git cmake build-essential libboost-dev libboost-system-dev libboost-filesystem-dev
    !pip3 install -qq lightgbm --install-option=--gpu