【LightGBM】在MacOS上安装LightGBM的GPU版

下载源码

git clone --recursive https://github.com/Microsoft/LightGBM
cd LightGBM
mkdir build ; cd build

安装cmakegcc,替换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

如果编译中出了问题,建议首先重新安装gcccmake,尤其是如果你之前就安装了这两个软件的话。