Python
每个版本都会将预构建的二进制 wheel 上传到 PyPI (Python Package Index)。支持的平台包括 Linux (x86_64, aarch64)、Windows (x86_64) 和 MacOS (x86_64, Apple Silicon)。
# Pip 21.3+ is required
pip install xgboost
如果您遇到权限错误,可能需要使用 --user 标志运行命令或使用 virtualenv。
注意
Python 包的部分内容现在需要 glibc 2.28+
从 2.1.0 版本开始,XGBoost Python 包将以两种变体发布
manylinux_2_28:适用于 glibc 2.28 或更新版本的最新 Linux 发行版。此变体启用所有功能。
manylinux2014:适用于 glibc 版本低于 2.28 的旧 Linux 发行版。此变体不支持 GPU 算法或联邦学习。
pip 包管理器将根据您的系统自动选择正确的变体。
从 2025 年 5 月 31 日开始,我们将停止分发 manylinux2014 变体,并仅分发 manylinux_2_28 变体。我们做出此决定是为了使我们的 CI/CD 管道不再依赖于已终止支持的软件组件(如 CentOS 7)。我们强烈建议所有人迁移到最新的 Linux 发行版,以便使用未来版本的 XGBoost。
注意:如果您想在较旧的 Linux 发行版上使用 GPU 算法或联邦学习,您有两种选择
升级到 glibc 2.28+ 的最新 Linux 发行版。或
从源代码构建 XGBoost。
注意
Windows 用户需要安装 Visual C++ 可再发行组件
XGBoost 需要来自Visual C++ 可再发行组件的 DLL 才能运行,因此请务必安装它。例外:如果您已安装 Visual Studio,则您已经拥有必要的库,因此无需安装 Visual C++ 可再发行组件。
各平台二进制 wheel 的功能
平台
GPU
多节点多 GPU
Linux x86_64
✔
✔
Linux aarch64
✘
✘
MacOS x86_64
✘
✘
MacOS Apple Silicon
✘
✘
Windows
✔
✘
最小安装 (仅限 CPU)
使用 pip 的默认安装将安装完整的 XGBoost 包,包括对 GPU 算法和联邦学习的支持。
您可以选择安装 xgboost-cpu 以减小安装包的大小并节省磁盘空间。
pip install xgboost-cpu
xgboost-cpu 变体将大幅减小磁盘占用,但不提供某些功能,例如 GPU 算法和联邦学习。
目前,xgboost-cpu 包适用于 x86_64 (amd64) Linux 和 Windows 平台。
Conda
您可以使用 Conda 包管理器安装 XGBoost
conda install -c conda-forge py-xgboost
Conda 应该能够检测您机器上是否存在 GPU 并安装正确版本的 XGBoost。如果您遇到问题,请尝试明确指出版本
# CPU variant
conda install -c conda-forge py-xgboost=*=cpu*
# GPU variant
conda install -c conda-forge py-xgboost=*=cuda*
要在没有 NVIDIA GPU 的机器上强制安装 GPU 版本,请使用环境变量 CONDA_OVERRIDE_CUDA,如 conda 文档中的“管理虚拟包”所述。
export CONDA_OVERRIDE_CUDA="12.8"
conda install -c conda-forge py-xgboost=*=cuda*
您可以从以下链接安装 Conda:下载 conda-forge 安装程序。