本文介绍了在Google Collab上为LightGBM安装GPU支持的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
有人通过使用那里的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支持的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!