lightGBM gpu环境配置

推荐先看一手官方的Installation Guide.我用的是ubuntu 16.04,一些要求如下图:

 

 

 主要是OpenCL以及libboost两个环境的要求.

(1) OpenCL的安装.我这里之前安装了NVIDA-395的驱动以及CUDA 9.0,只需要在编译的时候给出对应的path就可以了(这部分后面编译的时候会有解释).可以用clinfo查看OpenCL的信息.

Lenovo-Rescuer-15ISK:~$ clinfo
Number of platforms                               1
  Platform Name                                   NVIDIA CUDA
  Platform Vendor                                 NVIDIA Corporation
  Platform Version                                OpenCL 1.2 CUDA 9.1.84
  Platform Profile                                FULL_PROFILE
  Platform Extensions                             cl_khr_global_int32_base_atomics cl_khr_global_int32_extended_atomics cl_khr_local_int32_base_atomics cl_khr_local_int32_extended_atomics cl_khr_fp64 cl_khr_byte_addressable_store cl_khr_icd cl_khr_gl_sharing cl_nv_compiler_options cl_nv_device_attribute_query cl_nv_pragma_unroll cl_nv_copy_opts cl_nv_create_buffer
  Platform Extensions function suffix             NV

  Platform Name                                   NVIDIA CUDA
Number of devices                                 1
  Device Name                                     GeForce GTX 960M
  Device Vendor                                   NVIDIA Corporation
  Device Vendor ID                                0x10de
  Device Version                                  OpenCL 1.2 CUDA
  Driver Version                                  390.87
  Device OpenCL C Version                         OpenCL C 1.2 

(2)libboost安装.

> 从https://dl.bintray.com/boostorg/release/1.69.0/source/boost_1_69_0.tar.gz下载对应的压缩包

> 解压缩 tar -xzvf boost_1_69_0.tar.gz

> 切换到对应的文件目录下 sudo ./bootstrap.sh

> 根据提示 sudo ./b2 install 这样头文件就被默认安装在/usr/local/include头文件下,库文件就被默认安装在/usr/local/lib下

上述两个依赖搞定之后,安装官方给的安装方式(linux)如下:

git clone --recursive https://github.com/microsoft/LightGBM ; cd LightGBM
mkdir build ; cd build
cmake -DUSE_GPU=1 ..
# if you have installed NVIDIA CUDA to a customized location, you should specify paths to OpenCL headers and library like the following:
# cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ ..
make -j4

注意,如果已经安装过cuda的要指出OpenCL所在的位置,之后正常编译就好了.

 

 

 

   

 

转载于:https://www.cnblogs.com/z1141000271/p/11510776.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
配置LightGBM以使用GPU,您可以按照以下步骤进行操作: 1. 首先,确保您已经安装了所需的依赖项。根据引用和引用,您需要安装Git和CMake。 2. 克隆LightGBMGitHub存储库。您可以使用以下命令克隆存储库: ``` git clone --recursive https://github.com/Microsoft/LightGBM ``` 3. 进入LightGBM目录并创建一个build文件夹: ``` cd LightGBM mkdir build ``` 4. 进入build文件夹并运行CMake来配置LightGBM以使用GPU。根据您的CUDA安装位置,您可能需要指定OpenCL头文件和库的路径。运行以下命令: ``` cd build cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ .. ``` 5. 运行make命令以编译LightGBM并安装GPU支持: ``` make -j4 ``` 通过按照上述步骤进行操作,您应该能够成功配置LightGBM以使用GPU。请确保您已经正确安装了所需的依赖项,并根据您的CUDA安装位置进行相应的调整。引用和引用提供了更详细的安装说明。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [lightgbm安装(GPU)+用例](https://blog.csdn.net/u012969412/article/details/71433960)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *2* [lightGBM gpu环境配置](https://blog.csdn.net/diaoqi6581/article/details/101999630)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *3* [linux安装lightGBMGPU版本)](https://blog.csdn.net/weixin_39517520/article/details/116994501)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 33.333333333333336%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值