windows安装gpu版本的lightgbm,并添加进python环境
安装gpu版本的lightgbm,并且将其添加进python环境中
首先参照安装指南
编译安装完成后,检查LightGBM\Release下是否出现这里的DLL和exe

接着,我们在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是核显,不是独显

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

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

浙公网安备 33010602011771号