build-rocm-image

Build a ROCm development Docker image on a remote host via SSH.

260|104|Updated Nov 23, 2025
One-click install
npx skills add https://github.com/ROCm/FlyDSL --skill build-rocm-image-rocm
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: build-rocm-image
Source: https://github.com/ROCm/FlyDSL/tree/main/.claude/skills/build-rocm-image
Command: npx skills add https://github.com/ROCm/FlyDSL --skill build-rocm-image-rocm

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill automates the process of building a ROCm development Docker image on a remote host, consolidating multiple components (rocprofv3, vllm, aiter, FlyDSL, and a custom triton) into a single reproducible workflow.

Core Features & Use Cases

  • Generate a Dockerfile on the remote host based on a defined base image (rocm/vllm-dev:nightly) and customized components.
  • Clone and install required components (triton from rocm-maxnreg-support-v35, vllm from ps_pa branch, aiter from main, FlyDSL from develop) and verify the installations.
  • Validate the build by reporting versions and branch info, and provide instructions for running containers with GPU access.

Quick Start

SSH into the target host and run the build-rocm-image workflow with the hostname as the argument.

Frequently Asked Questions about build-rocm-image

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I build a ROCm Docker image on a remote host with vllm and triton?

To build a ROCm Docker image on a remote host, you need SSH access and a scripted workflow that generates a remote Dockerfile from the rocm/vllm-dev:nightly base image, clones required components, and performs the build.

Can I install FlyDSL and rocprofv3 together in a remote ROCm dev environment?

Yes, you can install FlyDSL and rocprofv3 together by automating a Docker build workflow that clones FlyDSL from the develop branch and consolidates it with rocprofv3 and aiter into a single reproducible remote ROCm dev environment.

What prerequisites are needed to set up a remote ROCm development environment?

Setting up a remote ROCm development environment requires SSH access to the target host and the rocm/vllm-dev:nightly base image to serve as the foundation for the scripted Docker build process.

How do I verify installed versions and branches after building a ROCm dev image?

You verify installed versions and branches after building a ROCm dev image by running scripted validation steps that report the specific branch info for components like triton and vllm, ensuring the reproducible build succeeded.

Does building a ROCm dev image on a remote host require a specific triton branch?

Building this ROCm dev image requires a custom triton cloned specifically from the rocm-maxnreg-support-v35 branch, alongside vllm from the ps_pa branch, to ensure precise component versioning.

What is the best way to run a ROCm container with GPU access after a remote build?

The best way to run a ROCm container with GPU access after a remote build is to follow the provided post-build instructions, which validate the image and detail the specific Docker run commands needed for GPU access.