返回介绍

数学基础

统计学习

深度学习

工具

Scala

一、安装

发布于 2023-07-17 23:38:23 字数 2951 浏览 0 评论 0 收藏 0

  1. 安装步骤:

    • 下载源码,编译lib_lightgbm.so (对于windows 则是 lib_lightgbm.dll

      
      git clone --recursive https://github.com/Microsoft/LightGBM
      # 对于 windows 用户,需要执行:
      #   git submodule init
      #   git submodule update
      cd LightGBM
      mkdir build
      cd build
      cmake ..
      make -j4

      如果需要支持GPU,则执行cmake 时指定:

      
      
      xxxxxxxxxx
      cmake -DUSE_GPU=1 .. # 如果你已经安装了 NVIDIA OpenGL,则使用: # cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ ..

      如果想要protobuf 来保存和加载模型,则先安装protobuf c++版本,然后使用:

      
      
      xxxxxxxxxx
      cmake -DUSE_PROTO=ON ..

      当前版本不支持该功能

    • 安装python 支持

      • 其中 --precompile 指示:已经编译过,不需要再次编译
      
      
      xxxxxxxxxx
      cd python-package sudo python setup.py install --precompile
  2. 直接pip 安装:

    
    
    xxxxxxxxxx
    pip install lightgbm pip install --no-binary :all: lightgbm #从源码编译安装 pip install lightgbm --install-option=--mpi #从源码编译安装 MPI 版本 pip install lightgbm --install-option=--gpu #从源码编译安装 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" #从源码编译安装,指定配置 #可选的配置有: # boost-root # boost-dir # boost-include-dir # boost-librarydir # opencl-include-dir # opencl-library

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
    我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
    原文