LightGBM在MacOS上安装LightGBM的GPU版
Posted zgljl2012
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LightGBM在MacOS上安装LightGBM的GPU版相关的知识,希望对你有一定的参考价值。
下载源码
git clone --recursive https://github.com/Microsoft/LightGBM
cd LightGBM
mkdir build ; cd build
安装cmake
和gcc
,替换AppleClang
Clang
苹果自带编译器LLVM
的前端,不支持LightGBM的编译,所以需要将其替换为gcc
。
brew list cmake
brew list gcc
运行上述命令,如果有则不需安装,否则,执行:
brew install cmake
brew install gcc
同时,大家可以看到cmake和gcc的版本,我的是7
,然后,执行:
export CXX=g++-7 CC=gcc-7
将7替换为自己的版本号。
接下来是修改 MaxOS 的默认编译器:
vim ~/.bash_profile
在最后新增:
alias gcc='gcc-7'
alias cc='gcc-7'
alias g++='g++-7'
alias c++='c++-7'
然后:
source ~/.bash_profile
这样就替换完成了。当然,我这里默认大家使用的是 Bash,如果你已经换成了 Zsh,那么,要改的文件就是 ~/.zshrc
了。
编译
cmake -DUSE_GPU=1 ..
make -j4
如果编译中出了问题,建议首先重新安装gcc
和cmake
,尤其是如果你之前就安装了这两个软件的话。
以上是关于LightGBM在MacOS上安装LightGBM的GPU版的主要内容,如果未能解决你的问题,请参考以下文章