在 Google Collab 上为 LightGBM 安装 GPU 支持答案

【问题标题】：Installing GPU support for LightGBM on Google Collab在 Google Collab 上为 LightGBM 安装 GPU 支持
【发布时间】：2020-05-05 23:23:16
【问题描述】：

任何人在尝试使用 Google Collab 上的 Notebooks 为 lightgbm 安装 GPU 支持时很幸运？

【问题讨论】：

标签： python gpu google-colaboratory lightgbm

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

【讨论】：

此解决方案在 Google Colab 上有效（2020 年 9 月）...另一个有一些问题 - 我为此苦苦挣扎了一段时间，因为它是 lightgbm 安装文档中建议的方法。我想强调一个事实，即您必须卸载现有的 lightgbm 包。 Colab 还会缓存包，因此您可能需要重新启动运行时以确保正在导入新安装的 lightgbm，而不是旧的。

【解决方案2】：

确保您正确执行了安装步骤

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

在此之后，您必须执行 LightGBM 文件夹中的 setupfile

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

完成后，一切就绪。 ps：确保你已经安装了cmake，如果不只是

!pip install cmake

【讨论】：

你克隆了第一个 LightGbm 并再次克隆？
我认为您不需要再次克隆它。我的错！

【解决方案3】：

大部分内容都遵循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

【讨论】：

不要忘记“运行时”->“更改运行时类型”->选择“GPU”作为硬件加速器。检查 GPU 是否存在并查看正在运行的作业：!nvidia-smi