Groot Rocker Extensions

Extensions for groot_rocker.

• bind: mount host dir(s) at specified location(s) inside the container
• development_environment: system dependencies and env variables for development
• git: transfer the user's git configuration across to the container
• named_prompt: a coloured @ prompt
• nvidia: enable nvidia
• ssh: install ssh dependencies, transfer the ssh agent into the container
• user: replicate the script's user in the image (useful for avoiding permission problems)
• work_directory: specify the starting directory on entry into the container (use with --bind)

Prerequisites

Currently tested and used on Ubuntu 18.04 with Python v3.

Docker & NVidia

# Docker
$ sudo apt install docker.io

# Nvidia Docker 2 - https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker
#   Not required, but fetch if you want to enable the nvidia extension in groot_rocker_extensions
$ HOST_DISTRIBUTION=$(. /etc/os-release; echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/${HOST_DISTRIBUTION}/nvidia-docker.list | \
$ sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update
$ sudo apt-get -q -y install nvidia-docker2 > /dev/null
$ sudo systemctl restart docker
# Test
$ docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi

Installation

From PyPi: pip3 -U install groot_rocker

In a Virtual Environment:

$ git clone https://github.com/stonier/groot_rocker.git ./groot_rocker
$ cd groot_rocker
$ . ./venv.bash
$ groot-rocker --help
# Build a .deb
$ make deb

From PPA:

$ sudo apt-add-repository ppa:d-stonier/snorriheim
$ sudo apt update
$ sudo apt install python3-groot-rocker-extensions
$ groot-rocker --help
$ groot-rocker-workspace --help

To test - try the workspace examples below.

Examples

Development Workspace

A customised image/container for your development workspace (foo):

# Full suite of options
$ groot-rocker-workspace --name foo --bind /mnt/mervin/workspaces/foo:/mnt/foo --work-directory /mnt/foo ubuntu:18.04

# If --bind has only 1 argument, --work-directory will be automagically configured.
# The previous command equivalent to:
$ groot-rocker-workspace --name foo --bind /mnt/mervin/workspaces/foo:/mnt/foo -- ubuntu:18.04

This will create both a named image (workspace:foo) and container (foo). The named image is useful merely for image management. The named container is very useful for subsequent interactions with the container. Note that the same command if executed again would fail since the container is persisting. To enter the container again:

$ docker container ls -a
CONTAINER ID IMAGE     COMMAND             CREATED    STATUS     NAMES
4db981f214b5 devel:foo "/bin/bash --login -i" 2 mins ago Exited (0) foo

$ docker container start -i foo

Launch a different development environment - bar:

$ groot-rocker-workspace --name bar --bind /mnt/mervin/workspaces/bar:/mnt/bar -- ubuntu:18.04

Ping between the two containers across the docker bridge:

# Foo
$ docker container start -i foo
$ ifconfig
eth0: flags=4163<UP,BROADCAST,RUNNING,MULTICAST>  mtu 1500
        inet 172.17.0.2  netmask 255.255.0.0  broadcast 172.17.255.255

# Bar
$ docker container start -i bar
$ ifconfig
eth0: flags=4163<UP,BROADCAST,RUNNING,MULTICAST>  mtu 1500
        inet 172.17.0.3  netmask 255.255.0.0  broadcast 172.17.255.255
$ ping 172.17.0.2
PING 172.17.0.2 (172.17.0.2) 56(84) bytes of data.
64 bytes from 172.17.0.2: icmp_seq=1 ttl=64 time=0.047 ms
$ ssh `whoami`@192.168.1.4  # replace this ip with your host's ip

The groot-rocker-workspace script is a convenience wrapper around the underlying groot-rocker command. Fully expanded:

$ groot-rocker \
  --development-environment \
  --user \
  --git \
  --ssh \
  --persistent \
  --image-name devel:foo \
  --container-name foo \
  --named-prompt \
  --bind /mnt/mervin/workspaces/foo:/mnt/foo \
  --work-directory /mnt/foo \
  ubuntu:18.04 \
  "/bin/bash --login -i"

Nvidia Benchmark

Be sure to install nvidia-docker 2.

Benchmarking tests:

# OpenGL
$ groot-rocker --nvidia --user --image-name groot:glmark2 ubuntu:18.04 "glmark2"

# Vulkan
$ groot-rocker --nvidia --user --image-name groot:vulkan ubuntu:20.04 "vkcube"

Nvidia Workspace

An nvidia enabled workspace - simply add --nvidia to the argument list for a development workspace:

$ groot-rocker-workspace \
  --name foo \
  --bind /mnt/mervin/workspaces/foo:/mnt/foo \
  --work-directory /mnt/foo \
  --nvidia \
  ubuntu:18.04

Once inside, test the nvidia performace with glmark2.