Install XGBoost with pip

2026-07-30

1 min read time

Applies to Linux

Prerequisites

To use XGBoost 3.2.0 on ROCm, you need the following prerequisites:

• ROCm version: 7.2.3, 7.1.1, 7.0.2

• Operating system: Ubuntu 24.04

• GPU platform: AMD Instinct™ MI355X (ROCm 7.2.3 only), MI325X, MI300X

• Python: 3.12, 3.11, 3.10

For a consistent and streamlined setup experience, it’s recommended to use a ROCm development environment Docker container. See Install AMD Finance for instructions.

Install using pip

Install the ROCm-enabled XGBoost library from the AMD-hosted PyPI (Python Package Index) repository.

ROCm 7.2.3

pip install amd_xgboost --extra-index-url=https://pypi.amd.com/rocm-7.2.3/simple

ROCm 7.1.1

pip install amd_xgboost --extra-index-url=https://pypi.amd.com/rocm-7.1.1/simple

ROCm 7.0.2

pip install amd_xgboost --extra-index-url=https://pypi.amd.com/rocm-7.0.2/simple

Verify your installation

Use pip show to verify your installation:

pip show -v amd_xgboost

Example output

Name: amd_xgboost
Version: 3.2.0
Summary: XGBoost Python Package
... [output truncated]

After installing XGBoost, import and use the library. For example:

import xgboost as xgb
print(f"XGBoost version: {xgb.__version__}")

# Create sample data
from sklearn.datasets import make_classification
X, y = make_classification(n_samples=1000, n_features=10, random_state=42)

# Test GPU training
dtrain = xgb.DMatrix(X, label=y)
params = {'tree_method': 'hist', 'device': 'cuda', 'max_depth': 3}
model = xgb.train(params, dtrain, num_boost_round=10)
print("GPU training successful!")