windows安装gpu版本的lightgbm,并添加进python环境

安装gpu版本的lightgbm,并且将其添加进python环境中
首先参照安装指南

https://lightgbm.cn/en/stable/Installation-Guide.html

编译安装完成后,检查LightGBM\Release下是否出现这里的DLL和exe
image

接着,我们在python环境中,pip install lightgbm

import lightgbm as lgb

# 创建一个简单的数据集
import numpy as np
X = np.random.rand(100, 10)
y = np.random.randint(0, 2, 100)

# 尝试使用GPU训练
params = {
    'device': 'gpu',
    'gpu_platform_id': 0,
    'gpu_device_id': 0,
    'objective': 'binary'
}

train_data = lgb.Dataset(X, label=y)
model = lgb.train(params, train_data, num_boost_round=10)
print("GPU训练完成")

先测试是否安装好了,这里发现,虽然安装好了,但是用的GPU是核显,不是独显
image

我们将编译好后的LightGBM\release\lib_lightgbm.dll下的文件拷贝覆盖到python环境的管理包下venv\Lib\site-packages\lightgbm\bin\lib_lightgbm.dll

image

那么接下来就用lightGBM去训练你的模型叭

posted @ 2025-09-01 11:39  yangzilu  阅读(140)  评论(0)    收藏  举报