TheRock/rocm-libraries/projects/hiptensor/docker/README.md

hipTensor Docker Images

Two Dockerfiles are provided, both targeting Ubuntu 24.04:

File	Description
Dockerfile.ubuntu24.prebuilt	Downloads a prebuilt ROCm tarball (nightly, stable, or latest)
Dockerfile.ubuntu24.fullbuild	Clones and builds ROCm from source (TheRock)

---

Prebuilt

Downloads a prebuilt ROCm tarball from a configurable source (THEROCK_URL_BASE). Faster to build; suitable for day-to-day development.

Default build (gfx94X, latest available nightly):

docker build -f Dockerfile.ubuntu24.prebuilt -t hiptensor:prebuilt .

THEROCK_GIT_TAG defaults to latest, which queries THEROCK_URL_BASE at build time, finds all tarballs matching the ASIC prefix, and downloads the highest version. ADHOCBUILD tarballs are excluded.

Select another ASIC :

docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_ASIC=gfx950 \
  -t hiptensor:prebuilt .

Pin to a specific nightly (format: <version>a<YYYYMMDD>):

docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_ASIC=gfx950 \
  --build-arg THEROCK_GIT_TAG=7.13.0a20260401 \
  -t hiptensor:prebuilt .

Use stable releases instead of nightlies:

docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_GIT_TAG=latest \
  --build-arg THEROCK_URL_BASE=https://repo.amd.com/rocm/tarball/ \
  -t hiptensor:prebuilt .

Pin to a specific stable release (format: <version>):

docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_GIT_TAG=7.12.0 \
  --build-arg THEROCK_URL_BASE=https://repo.amd.com/rocm/tarball/ \
  -t hiptensor:prebuilt .

Available tarball sources (THEROCK_URL_BASE):

Source	URL
Nightly builds (default)	https://rocm.nightlies.amd.com/tarball/
Stable releases	https://repo.amd.com/rocm/tarball/
Prereleases (QA)	https://rocm.prereleases.amd.com/tarball/
Dev builds	https://rocm.devreleases.amd.com/tarball/

See the TheRock releases page for details.

ASIC suffix note

The tarball suffix (-dcgpu, -dgpu, -all, or none) is derived automatically from THEROCK_ASIC:

ASIC(s)	Auto-selected suffix
gfx94X, gfx90X, gfx950	-dcgpu
gfx101X, gfx103X	-dgpu
gfx110X, gfx120X	-all
gfx900, gfx906, gfx908, gfx90a, gfx1150–gfx1153	(none)

Use THEROCK_PREBUILT_ID only if you need a non-default suffix (e.g. gfx110X-dgpu instead of gfx110X-all):

docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_PREBUILT_ID=gfx110X-dgpu \
  -t hiptensor:prebuilt .

Providing a full tarball URL or local path

If the tarball source uses a non-standard naming convention, bypass all auto-detection by providing the complete download URL via THEROCK_TARBALL_URL. All other THEROCK_* arguments are ignored when this is set:

docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_TARBALL_URL=https://repo.amd.com/rocm/tarball/therock-dist-linux-gfx950-dcgpu-7.12.0.tar.gz \
  -t hiptensor:prebuilt .

THEROCK_TARBALL_URL also accepts local paths resolved relative to the Docker build context. Use a file:/// prefix for an absolute path within the build context, or a plain relative path:

# Absolute path within the build context
docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_TARBALL_URL=file:///therock-dist-linux-gfx94X-dcgpu-7.12.0.tar.gz \
  -t hiptensor:prebuilt .

# Relative path within the build context
docker build -f Dockerfile.ubuntu24.prebuilt \
  --build-arg THEROCK_TARBALL_URL=tarballs/therock-dist-linux-gfx94X-dcgpu-7.12.0.tar.gz \
  -t hiptensor:prebuilt .

The build context is the directory passed to docker build (. by default). The tarball must be inside it — Docker cannot access files outside the build context, and passing a large directory as the context will cause Docker to transfer its entire contents to the daemon before the build starts.

If the tarball lives outside the current directory, use a temporary build context containing only the tarball:

TARBALL=/path/to/therock-dist-linux-gfx94X-dcgpu-7.12.0.tar.gz
CTX=$(mktemp -d)
cp "$TARBALL" "$CTX/"
docker build \
  -f "$(pwd)/Dockerfile.ubuntu24.prebuilt" \
  --build-arg THEROCK_TARBALL_URL=$(basename "$TARBALL") \
  -t hiptensor:prebuilt \
  "$CTX"
rm -rf "$CTX"

---

Fullbuild

Clones TheRock from GitHub and builds ROCm from source. Takes significantly longer; useful when a prebuilt tarball is unavailable or when custom build options are needed.

Default build (gfx94X, latest default branch):

docker build -f Dockerfile.ubuntu24.fullbuild -t hiptensor:fullbuild .

Specify a different ASIC and commit hash:

docker build -f Dockerfile.ubuntu24.fullbuild \
  --build-arg THEROCK_ASIC=gfx950 \
  --build-arg THEROCK_GIT_HASH=abc123def456 \
  -t hiptensor:fullbuild .

Debug build using 32 parallel jobs:

docker build -f Dockerfile.ubuntu24.fullbuild \
  --build-arg THEROCK_BUILD_MODE=Debug \
  --build-arg BUILD_JOBS=32 \
  -t hiptensor:debug .

Build with a custom TheRock CMake preset:

docker build -f Dockerfile.ubuntu24.fullbuild \
  --build-arg THEROCK_BUILD_MODE=Preset \
  --build-arg THEROCK_BUILD_PRESET=linux-debug-package \
  -t hiptensor:custom-preset .

Build mode options (THEROCK_BUILD_MODE):

Mode	Description
Release (default)	Standard optimized build
Debug	Debug symbols, no optimization
Preset	Use a named TheRock CMake preset (THEROCK_BUILD_PRESET)

---

Running a Container

Mount your current directory into /workspace:

docker run -itd --name my-container --rm --network=host --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
  -v $(pwd):/workspace \
  -v $HOME/.ssh:/root/.ssh/ \
  hiptensor:prebuilt
docker exec -it --privileged my-container /bin/bash