Install AMD Finance

Install AMD Finance#

2026-07-16

2 min read time

Applies to Linux

This page provides brief guidance and recommendations on setting up a ROCm-enabled environment for financial computing workloads. This includes pulling and running prebuilt ROCm Docker images for supported Ubuntu versions and installing AMD Finance libraries such as XGBoost, LightGBM, and ThunderGBM.

Install ROCm#

To get up and running quickly, a prebuilt ROCm-enabled container is recommended. The easiest way is to use the offical ROCm Docker images from Docker Hub. See Docker images in the ROCm ecosystem for more information.

Pull a ROCm dev Docker image with a supported configuration (see Docker Hub to browse available images). For example:
ROCm 7.1.1
docker pull rocm/dev-ubuntu-24.04:7.1.1-complete
Note

LightGBM and ThunderGBM are not yet supported on ROCm 7.1.1.

See rocm/dev-ubuntu-24.04:7.0.2-complete on Docker Hub.
ROCm 7.0.2
docker pull rocm/dev-ubuntu-24.04:7.0.2-complete
See rocm/dev-ubuntu-24.04:7.0.2-complete on Docker Hub.
ROCm 6.4.4
docker pull rocm/dev-ubuntu-24.04:6.4.4-complete
See rocm/dev-ubuntu-24.04:6.4.4-complete on Docker Hub.

Launch the Docker container.

ROCm 7.1.1

docker run -it \
    --cap-add=SYS_PTRACE \
    --ipc=host \
    --privileged=true \
    --shm-size=128GB \
    --network=host \
    --device=/dev/kfd \
    --device=/dev/dri \
    --group-add video \
    -v $HOME:$HOME \
    --name rocm711 \
    rocm/dev-ubuntu-24.04:7.1.1-complete

Note

LightGBM and ThunderGBM are not yet supported on ROCm 7.1.1.

ROCm 7.0.2

docker run -it \
    --cap-add=SYS_PTRACE \
    --ipc=host \
    --privileged=true \
    --shm-size=128GB \
    --network=host \
    --device=/dev/kfd \
    --device=/dev/dri \
    --group-add video \
    -v $HOME:$HOME \
    --name rocm7 \
    rocm/dev-ubuntu-24.04:7.0.2-complete

ROCm 6.4.4

docker run -it \
    --cap-add=SYS_PTRACE \
    --ipc=host \
    --privileged=true \
    --shm-size=128GB \
    --network=host \
    --device=/dev/kfd \
    --device=/dev/dri \
    --group-add video \
    -v $HOME:$HOME \
    --name rocm6 \
    rocm/dev-ubuntu-24.04:6.4.4-complete

To learn about ROCm Docker images, see Redirecting….