为了安装LightGBM-GPU的版本,在网上兜兜转转搜索了两天,在此记录自己成功安装的经历,也帮助更多想要在Windows环境下安装LightGBM-GPU的人避免踩坑
中文文档链接安装指南 · LightGBM 中文文档 · 看云
1、安装VStudio,这个建议自己在CSDN上搜索安装方法,在此不再赘述
2、安装Boost C++ Libraries文件,这里推荐下载.exe后缀的
我电脑上安装的VStudio2022,所以安装的是msvc14.3,如果对于自己应该下载哪个版本有所疑问,可以百度VStudio对应的MSVC版本,不过网上都说微软的东西是向下兼容的,我没有尝试过,因此不打保票。
安装完毕之后不需要编译。
3、安装CMake
推荐下载.MSI后缀的版本,可以直接安装,具体安装步骤百度CMAKE安装教程
安装完毕之后在cmd中输入cmake -version,如果安装成功,会有版本输出。
4、完成以上步骤之后,随便找个盘,用cmd输入
git clone --recursive https://github.com/microsoft/LightGBM
5. 进入lightGBM文件夹,在文件导航栏中输入cmd
直接在导航栏中输入cmd
激活你想要安装到的conda虚拟环境下
conda activate envname
conda install m2-base
在m2-base安装完毕之后,不要关闭这个cmd,接着输入
sh ./build-python.sh install --gpu
安装成功的话,命令行中会有一行绿色的成功安装提示
6、进入这个虚拟环境,测试是否安装成功
import lightgbm as lgb
import numpy as np
def check_gpu_support():
data = np.random.rand(50, 2)
label = np.random.randint(2, size=50)
print(label)
train_data = lgb.Dataset(data, label=label)
params = {'num_iterations': 1, 'device': 'gpu','gpu_platform_id':0, 'gpu_device_id':0}
# try:
gbm = lgb.train(params, train_set=train_data)
# print("GPU True !!!")
# except Exception as e:
# print("GPU False !!!")
if __name__ == '__main__':
check_gpu_support()
如果出现以下提示,说明安装成功