search

NeroCube / bookmark

Place some learning resources
0 stars 0 forks source link

LGBM GPU install #309

Open NeroCube opened 3 years ago

NeroCube commented 3 years ago

pip install

# default
pip install lightgbm

# MPI version
pip install lightgbm --install-option=--mpi

# GPU version
pip install lightgbm --install-option=--gpu

source code

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

# open MPI
# cmake -DUSE_MPI=ON ..

# GPU version
# cmake -DUSE_GPU=1 ..
make -j4