在 Google Colabratory 上使用 GPU 获取 LightGBM/ LGBM 运行
Get LightGBM/ LGBM run with GPU on Google Colabratory
我经常在 Google Colabratory 上 运行 LGBM，我刚刚发现此页面说 LGBM 默认设置为 CPU，因此您需要先进行设置。
https://medium.com/@am.sharma/lgbm-on-colab-with-gpu-c1c09e83f2af
所以我执行了页面上推荐的代码或Whosebug上推荐的其他一些代码如下，
!git clone --recursive https://github.com/Microsoft/LightGBM
%cd LightGBM
!mkdir build
%cd build
!cmake ../../LightGBM
!make -j4
!git clone --recursive https://github.com/Microsoft/LightGBM.git
%cd LightGBM/python-package
!python3 setup.py install --gpu
!pip install cmake
但我仍然遇到错误
GPU Tree Learner was not enabled in this build.Please recompile with CMake option -DUSE_GPU=1
我该如何解决？
谢谢！
宁运行后
! git clone --recursive https://github.com/Microsoft/LightGBM
您可以运行这个 oneliner，它将在 colab 中启用 GPU 构建和编译 LightGBM：
! cd LightGBM && rm -rf build && mkdir build && cd build && cmake -DUSE_GPU=1 ../../LightGBM && make -j4 && cd ../python-package && python3 setup.py install --precompile --gpu;
希望对您有所帮助 ;D
我经常在 Google Colabratory 上 运行 LGBM，我刚刚发现此页面说 LGBM 默认设置为 CPU，因此您需要先进行设置。
https://medium.com/@am.sharma/lgbm-on-colab-with-gpu-c1c09e83f2af
所以我执行了页面上推荐的代码或Whosebug上推荐的其他一些代码如下，
!git clone --recursive https://github.com/Microsoft/LightGBM
%cd LightGBM
!mkdir build
%cd build
!cmake ../../LightGBM
!make -j4
!git clone --recursive https://github.com/Microsoft/LightGBM.git
%cd LightGBM/python-package
!python3 setup.py install --gpu
!pip install cmake
但我仍然遇到错误
GPU Tree Learner was not enabled in this build.Please recompile with CMake option -DUSE_GPU=1
我该如何解决？ 谢谢！
宁运行后
! git clone --recursive https://github.com/Microsoft/LightGBM
您可以运行这个 oneliner，它将在 colab 中启用 GPU 构建和编译 LightGBM：
! cd LightGBM && rm -rf build && mkdir build && cd build && cmake -DUSE_GPU=1 ../../LightGBM && make -j4 && cd ../python-package && python3 setup.py install --precompile --gpu;
希望对您有所帮助 ;D