Install ThunderGBM with pip

Install ThunderGBM with pip#

2025-12-15

1 min read time

Applies to Linux

Prerequisites#

Before proceeding, ensure that you have installed a supported ROCm version, operating system, and Python version that are compatible with the ROCm Finance libraries. Verify that your system includes a supported AMD Instinct GPU For guidance, see ROCm finance installation prerequisites.

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

Install using pip#

Install the ROCm-enabled ThunderGBM library from the AMD-hosted PyPI repository.

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

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

Verify your installation#

Use pip show to verify your installation:

pip show -v amd_thundergbm
Example output
Name: amd_thundergbm
Version: 0.3.16
Summary: A Fast GBM Library on GPUs and CPUsi with ROCm support
Home-page: https://github.com/rocm/thundergbm
... [output truncated]

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

from thundergbm import TGBMClassifier
clf = TGBMClassifier()
Contents