XGBoost 설치
XGBoost 설치 (Installation Guide)
XGBoost는 몇몇 언어 바인딩에 대해 바이너리 패키지를 제공해요. 바이너리 패키지는 NVIDIA GPU가 있는 머신에서 GPU 알고리즘(device=cuda:0)을 지원해요. 단, 다중 GPU로 학습하는 건 Linux 플랫폼에서만 지원돼요. 안정 릴리스와 nightly 빌드가 모두 있으니 목적에 맞게 설치하면 돼요.
Python 안정 릴리스
각 릴리스마다 미리 빌드된 바이너리 휠이 PyPI에 올라가요. 지원 플랫폼은 Linux(x86_64, aarch64), Windows(x86_64), MacOS(x86_64, Apple Silicon)예요.
# Pip 21.3+ 필요
pip install xgboost
권한 오류가 나면 --user 플래그를 붙이거나 virtualenv를 써 보세요. 참고로 Python 패키지의 일부는 glibc 2.28+가 필요해요.
2.1.0부터 Python 패키지는 두 가지 변형으로 배포돼요.
manylinux_2_28: glibc 2.28 이상의 최신 Linux 배포판용. 모든 기능이 활성화돼 있어요.manylinux2014: glibc가 2.28보다 오래된 구형 Linux 배포판용. GPU 알고리즘이나 연합 학습(federated learning)을 지원하지 않아요.
pip은 시스템에 맞는 변형을 자동으로 골라줘요. 2025년 5월 31일부터는 manylinux2014 변형 배포를 중단하고 manylinux_2_28만 배포할 예정이에요. 구형 배포판에서 GPU 알고리즘이나 연합 학습을 쓰려면 최신 Linux로 업그레이드하거나 소스에서 직접 빌드해야 해요.
Windows 사용자는 Visual C++ Redistributable을 설치해야 XGBoost가 동작해요. Visual Studio가 설치돼 있다면 필요한 라이브러리를 이미 갖고 있으니 추가 설치가 필요 없어요.
플랫폼별 지원 기능
| 플랫폼 | GPU | Multi-Node-Multi-GPU |
|---|---|---|
| Linux x86_64 | ✓ | ✓ |
| Linux aarch64 | ✓ | ✗ |
| MacOS x86_64 | ✗ | ✗ |
| MacOS Apple Silicon | ✗ | ✗ |
| Windows | ✓ | ✗ |
Linux aarch64 휠에는 CUDA 지원이 포함돼 있어 최신 Jetson이나 Graviton 머신에서 pip install xgboost만으로 x86_64 휠과 같은 GPU 기능을 제공받을 수 있어요.
최소 설치 (CPU 전용)
기본 pip 설치는 GPU 알고리즘과 연합 학습 지원을 포함한 전체 패키지를 설치해요. 설치 크기를 줄이고 싶다면 CPU 전용 최소 설치 옵션을 선택할 수 있어요.
더 알아보기
- Python 패키지 소개 — Python 인터페이스
- XGBoost GPU 지원 — GPU 학습
- 소스에서 빌드 — 직접 빌드