在Google Collab上为LightGBM安装GPU支持 [英] Installing GPU support for LightGBM on Google Collab

问题描述

有人通过使用那里的Notebooks尝试在Google Collab上为lightgbm安装GPU支持吗?

Anyone got luck trying to install GPU support for lightgbm on Google Collab using the Notebooks there ?

推荐答案

确保正确执行了安装步骤

make sure you followed the installation steps correctly

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

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

after this you have to execute the setupfile in LightGBM folder

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

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

Once thats done , you're all set. ps: make sure you have cmake installed, if not just

!pip install cmake

这篇关于在Google Collab上为LightGBM安装GPU支持的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！