Windows环境Conda虚拟环境安装LightGBM-GPU教程

         为了安装LightGBM-GPU的版本,在网上兜兜转转搜索了两天,在此记录自己成功安装的经历,也帮助更多想要在Windows环境下安装LightGBM-GPU的人避免踩坑

中文文档链接安装指南 · LightGBM 中文文档 · 看云

1、安装VStudio,这个建议自己在CSDN上搜索安装方法,在此不再赘述

2、安装Boost C++ Libraries文件,这里推荐下载.exe后缀的

Boost C++ Libraries Files

        我电脑上安装的VStudio2022,所以安装的是msvc14.3,如果对于自己应该下载哪个版本有所疑问,可以百度VStudio对应的MSVC版本,不过网上都说微软的东西是向下兼容的,我没有尝试过,因此不打保票。

        安装完毕之后不需要编译。

3、安装CMake

Download CMake

推荐下载.MSI后缀的版本,可以直接安装,具体安装步骤百度CMAKE安装教程

安装完毕之后在cmd中输入cmake -version,如果安装成功,会有版本输出。

4、完成以上步骤之后,随便找个盘,用cmd输入

git clone --recursive https://github.com/microsoft/LightGBM

5. 进入lightGBM文件夹,在文件导航栏中输入cmd

直接在导航栏中输入cmd

激活你想要安装到的conda虚拟环境下

conda activate envname
conda install m2-base

在m2-base安装完毕之后,不要关闭这个cmd,接着输入

sh ./build-python.sh install --gpu

安装成功的话,命令行中会有一行绿色的成功安装提示

6、进入这个虚拟环境,测试是否安装成功

import lightgbm as lgb
import numpy as np

def check_gpu_support():
    data = np.random.rand(50, 2)
    label = np.random.randint(2, size=50)
    print(label)
    train_data = lgb.Dataset(data, label=label)
    params = {'num_iterations': 1, 'device': 'gpu','gpu_platform_id':0, 'gpu_device_id':0}
    # try:
    gbm = lgb.train(params, train_set=train_data)
    #     print("GPU True !!!")
    # except Exception as e:
    #     print("GPU False !!!")


if __name__ == '__main__':
    check_gpu_support()

如果出现以下提示,说明安装成功

  • 6
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
根据引用中的信息,安装LightGBMGPU版本可以尝试以下两种方法: 方法一: 使用以下命令安装LightGBMGPU版本: ``` sudo pip3 install lightgbm --install-option=--gpu ``` 方法二: 1. 克隆LightGBM的GitHub仓库: ``` git clone --recursive https://github.com/microsoft/LightGBM ``` 2. 进入克隆的仓库目录: ``` cd LightGBM ``` 3. 创建一个新的目录并进入: ``` mkdir build cd build ``` 4. 运行CMake配置命令编译LightGBM: ``` cmake -DUSE_GPU=1 .. ``` 如果你的NVIDIA CUDA安装在非默认位置,你需要指定OpenCL头文件和库的路径: ``` cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ .. ``` 5. 使用make命令进行编译: ``` make -j4 ``` 请注意,以上两种方法需要在命令行中执行,并确保你的系统已经正确安装了相关的依赖项。 如果你使用的是Anaconda环境,可以使用以下命令通过conda安装LightGBMGPU版本: ``` conda install -c conda-forge lightgbm ``` 安装完成后,你可以按照引用中的示例代码进行测试,以确认LightGBMGPU支持是否安装成功。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [Linux下安装LightGBM-GPU版本](https://blog.csdn.net/qq_35564813/article/details/104986194)[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 -- Light Gradient Boosting Machine](https://blog.csdn.net/CSDNhuaong/article/details/127352816)[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* [ubuntu20.04安装lightgbmgpu版本](https://blog.csdn.net/qq_37774098/article/details/120071007)[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、付费专栏及课程。

余额充值