有人试图使用笔记本电脑在Google Collab上为lightgbm安装GPU支持吗?
其中大部分内容都是遵循here提供的文档,并进行了两次小调整,以使其在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
确保正确遵循安装步骤
!git clone --recursive https://github.com/Microsoft/LightGBM
%cd LightGBM
!mkdir build
%cd build
!cmake ../../LightGBM
!make -j4
在此之后,您必须在LightGBM文件夹中执行setupfile
!git clone --recursive https://github.com/Microsoft/LightGBM.git
%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版本。