Docker Quickstart

Pulling a Docker Image

Your node comes with Docker Engine installed, so all Docker functionality should be available to you on your first connection. Begin by pulling the desired image:

You can verify that your image was properly pulled by running the following command and checking for your desired image:

If the pull was successful, your output should look similar to this:

Running a Docker Container

In order to run your Docker containers with GPU acceleration, you must mount the devices. For certain applications, you must also add the container to a group to utilize your GPUs.

Using the docker run Command

Here's an example command to mount the devices and configure the correct permissions:

The usage of each option is as follows:

• --device /dev/kfd

  • This command mounts the main compute interface to your container.

• --device /dev/dri

Using docker-compose

The following is an equivalent docker-compose to the command above:

To use it, create a docker-compose.yml file in any subdirectory, and within that subdirectory, run:

Verifying Setup

If done properly, the output of either the run or compose command should be similar to:

For other containers, to verify that your Docker container has access to your GPUs, run both rocm-smi and rocminfo. These commands will reveal information about the GPUs mounted to your container.