[英]Installing GPU support for LightGBM on Google Collab
有没有人尝试使用 Notebooks 在 Google Collab 上安装对 lightgbm 的 GPU 支持?
确保您正确地遵循了安装步骤
!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
很简单:只要运行
!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 上运行。
由于实例在使用 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
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.