标签：Info ... lightgbm LightGBM python gpu packages 10
- 环境
- Ubuntu 20.04.2
- Python 3.8.10
- Nvidia-smi driver 535.104.05
- cuda 12.2
- torch 1.13.1+cu116(torch cuda 11.6)
- 步骤
- 测试
-
> python3
Python 3.8.10 (default, Jul 29 2024, 17:02:10)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import lightgbm as lgb
>>> import numpy as np
>>> a1 = np.random.random((10,3))
>>> a2 = np.random.random(10)
>>> para = {'device_type': 'gpu'}
>>> model = lgb.train(para,train_set=lgb.Dataset(a1,a2))
[LightGBM] [Warning] There are no meaningful features which satisfy the provided configuration. Decreasing Dataset parameters min_data_in_bin or min_data_in_leaf and re-constructing Dataset might resolve this warning.
[LightGBM] [Info] This is the GPU trainer!!
[LightGBM] [Info] Total Bins 0
[LightGBM] [Info] Number of data points in the train set: 10, number of used features: 0
[LightGBM] [Info] Using GPU Device: NVIDIA H100 PCIe, Vendor: NVIDIA Corporation
[LightGBM] [Info] Compiling OpenCL Kernel with 16 bins...
1 warning generated.
1 warning generated.
... ...
1 warning generated.
1 warning generated.
[LightGBM] [Info] GPU programs have been built
[LightGBM] [Warning] GPU acceleration is disabled because no non-trivial dense features can be found
[LightGBM] [Info] Start training from score 0.482093
[LightGBM] [Warning] Stopped training because there are no more leaves that meet the split requirements
[LightGBM] [Warning] Stopped training because there are no more leaves that meet the split requirements
... ...
[LightGBM] [Warning] Stopped training because there are no more leaves that meet the split requirements
... ...
标签：Info,
...,
lightgbm,
LightGBM,
python,
gpu,
packages,
10
From： https://www.cnblogs.com/Song-Meow/p/18371521