Question

任何人都试图使用笔记本电脑在Google Collab上为lightgbm安装GPU支持吗？

Answer 1

大部分内容都是按照here提供的文档进行的，并进行了两次小调整，以使其在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

Answer 2

确保正确执行安装步骤

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

此后，您必须在LightGBM文件夹中执行设置文件

!git clone --recursive https://github.com/Microsoft/LightGBM.git
%cd LightGBM/python-package
!python3 setup.py install --gpu

一旦完成，就一切就绪。 ps：请确保您已经安装了cmake，

!pip install cmake

Answer 3

非常简单：只需运行

!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支持，并在lightgbm设置中添加'device':'gpu'。并且不要忘记先卸载不支持gpu版本的lightgbm版本。

在Google Collab上为LightGBM安装GPU支持

3 个答案: