Win10 平台下, LightGBM GPU 版本的安装www.jianshu.com
1). 下载并安装 Git for Win。
详细步骤可以参考这篇文章 Git安装教程(Windows安装超详细教程)
2). 下载并安装 CMake。
这个安装过程比较简单,不过有两点需要注意。
勾选中间那一项,对所有用户都添加到系统 path 路径中, 其他方面就一路 next 就可以了。
3). 下载并安装 VS Build Tools。
VS Build Tools 现在集成在 Visual Studio 之中,没有提供单独下载。Visual Studio 作为商业软,自然是收费的,但是微软提供了一个 Visual Studio Community 版本,可以免费使用,最新版为 Visual Studio Community 2017。
点击下载的 ".exe", 会出现一个安装选择面板,你可以勾选希望安装的功能,如果只是用于这次的编译和构造,只需要按下图进行勾选即可,然后安装即可。
我之前安装了vs2019后来又安装了 VS Build Tools
4). 下载安装相应的OpenCL
OpenCL 可以简单地理解为显卡的驱动,所以Intel、Nvidia和AMD需要安装不同的OpenCL。
因为之前装了tf的gpu版本,所以按照tf的教程里去安装gpu相应的事项就可以了
5). 下载安装 Boost Binaries
选择和 visual studio 版本对应的版本:
Visual Studio 2015:msvc-14.0-64.exe
Visual Studio 2017:msvc-14.1-64.exe
安装完成之后,会在C盘根目录下,出现 "C:\local\" 这个文件夹,把 "C:\local\boost_1_69_0\"和"C:\local\boost_1_69_0\lib64-msvc-14.1"添加到系统环境变量 (Note: 根据安装的Boost Binaries 版本不同,路径名称会有稍微不同)
6). 编译并构造源文件
这一步比较重要,之前就是这个boost binaries没有安装所以一直编译失败,进入链接之后安装最新的也就是最上面额msvc exe那个文件就可以了。在命令行中,输入:
Set BOOST_ROOT=C:/local/boost_1_69_0/ Set BOOST_LIBRARYDIR=C:/local/boost_1_64_0/lib64-msvc-14.2
这里的地址是我安装时候默认的地址,需要根据自己安装时候的设置自行设置
后续没有用可视化的界面来编译,不知道为啥可视化界面一直编译错误,而是使用了:https://blog.csdn.net/qq_31395299/article/details/82465176blog.csdn.netit clone --recursive microsoft/LightGBM
cd LightGBM
mkdir build
cd build
cmake -DCMAKE_GENERATOR_PLATFORM=x64 -DUSE_GPU=1 ..
cmake --build . --target ALL_BUILD --config Release
2、在命令提示符下,到LightGBM\python-package目录下执行
pip install lightgbm --install-option=--gpu
就可以了!!
如果要调用的话也很简单:
在参数里面设置device='gpu'就可以了,官网只给到了R的gpu调用的代码,不过参数放到python里面也是一样的。
params = {'num_leaves': 491,
'min_child_weight': 0.03454472573214212,
'feature_fraction': 1,#0.3797454081646243,
'bagging_fraction': 1,#0.4181193142567742,
'min_data_in_leaf': 106,
'objective': 'binary',
'max_depth': -1,
'learning_rate': 0.006883242363721497,
"boosting_type": "gbdt",
"bagging_seed": 11,
"metric": 'auc',
"verbosity": -1,
'reg_alpha': 0.3899927210061127,
'reg_lambda': 0.6485237330340494,
'random_state': 47,
"num_threads":-1,
"device":"gpu"
}
详细可参考GPU SDK Correspondence and Device Targeting Tablelightgbm.readthedocs.io
照葫芦画瓢就可以了。