verl on ROCm installation

verl on ROCm installation#

2026-03-03

2 min read time

Applies to Linux

System requirements#

To use verl 0.3.0.post0, you need the following prerequisites:

  • ROCm version: 6.2.0

  • Operating system: Ubuntu 20.04

  • GPU platform: AMD Instinct™ MI300X

  • PyTorch: 2.5.0

  • Python: 3.9

  • vLLM: 0.6.3

Install verl#

To install verl on ROCm, you have the following options:

Use a prebuilt Docker image with verl pre-installed#

The recommended way to set up a verl environment and avoid potential installation issues is with Docker. The tested, prebuilt image includes verl, PyTorch, ROCm, and other dependencies.

Prebuilt Docker images with verl configured for ROCm are available on Docker Hub.

  1. Pull the Docker image

    docker pull rocm/verl:verl-0.3.0.post0_rocm6.2_vllm0.6.3

  2. Launch and connect to the Docker container

    docker run --rm -it \
   --name rocm_verl \
   --device /dev/dri \
   --device /dev/kfd \
   --group-add video \
   --cap-add SYS_PTRACE \
   --security-opt seccomp=unconfined \
   --privileged \
   -p 8265:8265 \
   -v "$HOME/.ssh:/root/.ssh" \
   -v "$HOME:$HOME" \
   --shm-size 128G \
   -w "$PWD" \
   rocm/verl:verl-0.3.0.post0_rocm6.2_vllm0.6.3 \
   /bin/bash

Build your own Docker image#

  1. Clone the ROCm/verl repository

    git clone https://github.com/volcengine/verl.git -b v0.3.0.post0

  2. Build the Docker container using the Dockerfile in the verl/docker directory

    cd verl
docker build -f docker/Dockerfile.rocm -t my-rocm-verl .

  3. Launch and connect to the container

    docker run --rm -it \
   --name rocm_verl \
   --device /dev/dri \
   --device /dev/kfd \
   --group-add video \
   --cap-add SYS_PTRACE \
   --security-opt seccomp=unconfined \
   --privileged \
   -p 8265:8265 \
   -v "$HOME/.ssh:/root/.ssh" \
   -v "$HOME:$HOME" \
   --shm-size 128G \
   -w "$PWD" \
   my-rocm-verl \
   /bin/bash

    Note

    The --shm-size parameter allocates shared memory for the container. It can be adjusted based on your system’s resources.

Test the verl installation#

Once connected to the Docker container, verify that verl is installed:

pip list | grep verl
verl    0.3.0.post0        /app
Contents