logo

XGBoost 安装

wangzf / 2024-09-18


目录

Linux

  1. 下载源码
$ git clone --recursive https://github.com/dmlc/xgboost
  1. 编译 libxgboost.so
$ cd xgboost
$ make -j4
  1. 如果需要支持 GPU,则需要执行以下步骤
$ cd xgboost
$ mkdir build
$ cd build
$ cmake .. -DUSE_CUDA=ON
$ cd ..
make -j4
  1. 安装 Python 支持
$ cd python-package
$ sudo python setup.py intall

Windows

  1. 下载源码
$ git clone --recursive https://github.com/dmlc/xgboost
$ git submodule init
$ git submodule update
  1. 编译 libxgboost.dll
$ cd xgboost
$ make -j4
  1. 如果需要支持 GPU,则需要执行以下步骤
$ cd xgboost
$ mkdir build
$ cd build
$ cmake .. -DUSE_CUDA=ON
$ cd ..
make -j4
  1. 安装 Python 支持
$ cd python-package
$ python setup.py intall

macOS