Install ROCm-Simulation

Install ROCm-Simulation#

2025-12-15

2 min read time

Applies to Linux

This topic provides brief guidance and recommendations on setting up a ROCm-enabled environment for simulation workloads. This includes pulling and running a prebuilt ROCm Docker image for supported Ubuntu versions and installing components.

System requirements#

The ROCm-Simulation components are both supported on AMD Instinct™ MI325X and MI300X GPUs.

To use both Taichi Lang and GSplat, you need the following prerequisites:

  • ROCm version: 7.0.0

  • Operating system: Ubuntu 22.04, 24.04

  • PyTorch: 2.8 (ROCm-enabled)

  • Python: 3.12.3, 3.10.12

Note

If you are only installing Taichi Lang or GSplat, see the ROCm-Simulation compatibility matrix page.

Install ROCm-Simulation#

After confirming your system meets the supported hardware and software configurations, follow these steps:

  1. Install the supported ROCm version. To do this, pull a ROCm dev Docker image with a supported configuration (see Docker Hub to browse available images).

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

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

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

    See rocm/dev-ubuntu-22.04:7.0-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-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-complete

  3. Install each ROCm-Simulation component. The installation instructions for each component can be found as follows: