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()

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

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值