我经常在 Google Colabatory 上运行 LGBM,我刚刚发现这个页面说 LGBM 默认设置为 CPU,所以你需要先设置。
https://medium.com/@am.sharma/lgbm-on-colab-with-gpu-c1c09e83f2af
所以我执行了页面上推荐的代码或stackoverflow上推荐的其他一些代码,如下所示,
!git clone --recursive https://github.com/Microsoft/LightGBM
%cd LightGBM
!mkdir 构建
%cd 构建
!cmake ../../LightGBM
!make -j4
!git clone --recursive https://github.com/Microsoft/LightGBM.git
%cd LightGBM/python 包
!python3 setup.py install --gpu
!pip 安装 cmake
但我仍然收到错误
此版本中未启用 GPU Tree Learner。请使用 CMake 选项 -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
关于python - 在 Google Colaboratory 上使用 GPU 运行 LightGBM/LGBM,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58707252/