LightGBM

LightGBM 是一个基于决策树算法的快速、分布式、高性能的框架，用于排名、分类和许多其他机器学习任务。

LightGBM 可以使用CPU、GPU，其中 GPU 分为使用 opencl 的版本（可以使用 AMD、NVIDIA 的GPU，参数为 device='gpu'）、CUDA 版本（只能使用 NVIDIA GPU，参数为 device='cuda'），CUDA 版本比 GPU 版速度更快。使用时注意 device 参数设置，否则会调用失败。

项目地址 https://github.com/microsoft/LightGBM

官方文档 https://lightgbm.readthedocs.io/

中文文档 https://lightgbm.cn/

cuda 版 python 包安装¶

$ module load Python/3.11.4 GCC/9.4.0
$ export CC=`which gcc`
$ export CXX=`which g++`
$ pip install --prefix=/path/to/LightGBM/ lightgbm --no-binary lightgbm --config-settings=cmake.define.USE_CUDA=ON

集群上调用

$ module load LightGBM/4.6.0-cuda

测试¶

测试代码

import lightgbm as lgb
import numpy as np
# 测试 GPU 支持
try:
    params = {'device': 'cuda', 'verbosity': -1, 'objective': 'regression'}
    X = np.random.rand(100, 10)
    y = np.random.rand(100)
    model = lgb.train(params, lgb.Dataset(X, y), num_boost_round=1)
    print('GPU 测试通过')
except Exception as e:
    print(f'GPU 测试失败: {e}')

测试

$ module load LightGBM/4.6.0-cuda
$ python lightgbm.py 
GPU 测试通过

本文阅读量次
本站总访问量次