dmikushin / miopen-docker

Build the specified commit of MIOpen from source in a Docker container based on ROCm-docker

MIOpen Docker Container

Build the specified commit of MIOpen from source in a Docker container based on ROCm-docker.

The host Linux system must provide a compatible ROCm driver and a supported GPU.

The following setup has been tested:

• GPU: Radeon RX Vega gfx900
• Host: rocm-dev 4.5.0.40500-56
• Container: rocm-dev 5.0.1.50001-59

Upon the startup, the container replicates the current system user as a container user with exactly same groups and ids. This is one reliable way to allow GPU usage from within the container on behalf of non-priviledged user.

Usage

1. Specify the commit in build.sh
2. Execute ./build.sh

Building

docker build -t rocm/miopen-docker .

Running

The following commands assume that the current host system user is permitted to use the GPU.

• Run interactively from the command line:

docker run --rm -it --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined --env ID="$(id)" rocm/miopen-docker

• Run a background service with docker-compose, and connect to it when needed:

./compose.sh
docker-compose exec miopen-docker sudo -i -u $(whoami)