Install ROCm-Finance

Install ROCm-Finance#

2025-12-15

3 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 ROCm-Finance libraries such as 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.

  1. Pull a ROCm dev Docker image with a supported configuration (see Docker Hub to browse available images). For example:

    docker pull rocm/dev-ubuntu-24.04:7.0.2-complete
    

    See rocm/dev-ubuntu-24.04:7.0.2-complete on Docker Hub.

    docker pull rocm/dev-ubuntu-22.04:7.0.2-complete
    

    See rocm/dev-ubuntu-22.04:7.0.2-complete on Docker Hub.

    docker pull rocm/dev-ubuntu-24.04:6.4.4-complete
    

    See rocm/dev-ubuntu-24.04:6.4.4-complete on Docker Hub.

    docker pull rocm/dev-ubuntu-22.04:6.4.4-complete
    

    See rocm/dev-ubuntu-22.04:6.4.4-complete on Docker Hub.

  2. Launch the Docker container.

    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
    
    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-22.04:7.0.2-complete
    
    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
    
    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-22.04:6.4.4-complete
    

To learn about ROCm Docker images, see Running ROCm Docker containers.

Install ROCm-Finance libraries#

Get started using ROCm for the finance domain. To install ROCm-Finance libraries, see the following resources. For compatibility information, see the ROCm-Finance compatibility matrix.