initial role

Author: tibroc

.gitignore

@@ -0,0 +1,3 @@
+.vscode
+venv
+env

LICENSE

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2026, virtUOS
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

README.md

@@ -0,0 +1,60 @@
+# Ansible Role: nvidia_container_toolkit
+
+Installs and configures the [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/overview.html) for Docker on RPM-based systems.
+
+## What it does
+
+1. Adds the official NVIDIA Container Toolkit RPM repository.
+2. Installs the `nvidia-container-toolkit` package.
+3. Configures `/etc/docker/daemon.json` with the NVIDIA runtime.
+4. Restarts Docker when the daemon configuration changes.
+
+## Requirements
+
+- RPM-based OS (RHEL, Rocky, AlmaLinux, CentOS, Fedora)
+- Docker already installed
+- NVIDIA GPU drivers already installed on the host
+
+## Role Variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `nvidia_container_toolkit_repo_url` | `https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo` | URL to the NVIDIA Container Toolkit RPM repository file. |
+| `nvidia_docker_default_runtime` | `nvidia` | The default container runtime for Docker. |
+
+## Dependencies
+
+- nvidia_driver_cuda
+- elan.docker_install
+
+## Example Playbook
+
+### Basic usage
+
+```yaml
+- hosts: gpu_nodes
+  roles:
+    - nvidia_driver_cuda
+    - elan.docker_install
+    - nvidia_container_toolkit
+```
+
+### With custom variables
+
+```yaml
+- hosts: gpu_nodes
+  roles:
+    - role: nvidia_container_toolkit
+      vars:
+        nvidia_docker_default_runtime: "runc"
+```
+
+## Handlers
+
+- **Restart Docker:** Triggered when /etc/docker/daemon.json is modified.
+
+## Important Notes
+
+- This role overwrites /etc/docker/daemon.json. If you have existing Docker daemon configuration, merge it into the templates/daemon.json.j2 template.
+- This role does not install GPU drivers. Make sure drivers are present before running it.
+- This role only supports RPM-based distributions. For Debian/Ubuntu, the repository setup task needs to be adapted.

defaults/main.yml

@@ -0,0 +1,3 @@
+---
+nvidia_container_toolkit_repo_url: "https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo"
+nvidia_docker_default_runtime: "nvidia"

handlers/main.yml

@@ -0,0 +1,5 @@
+---
+- name: Restart Docker
+  ansible.builtin.systemd:
+    name: docker
+    state: restarted

meta/main.yml

@@ -0,0 +1,12 @@
+---
+galaxy_info:
+  role_name: nvidia_container_toolkit
+  namespace: uos
+  author: Timo Nogueira Brockmeyer
+  description: Install nvidia container toolkit
+  company: Osnabrück University
+  license: BSD-3-Clause
+  min_ansible_version: "0.1"
+dependencies:
+  - elan.docker_install
+  - nvidia_driver_cuda

tasks/main.yml

@@ -0,0 +1,23 @@
+---
+- name: Add NVIDIA Container Toolkit repository
+  ansible.builtin.get_url:
+    url: "{{ nvidia_container_toolkit_repo_url }}"
+    dest: /etc/yum.repos.d/nvidia-container-toolkit.repo
+    mode: "0644"
+
+- name: Install NVIDIA Container Toolkit
+  ansible.builtin.package:
+    name: nvidia-container-toolkit
+
+- name: Create Docker daemon configuration directory
+  ansible.builtin.file:
+    path: /etc/docker
+    state: directory
+    mode: "0755"
+
+- name: Configure Docker daemon with NVIDIA runtime support
+  ansible.builtin.template:
+    src: daemon.json.j2
+    dest: /etc/docker/daemon.json
+    mode: "0644"
+  notify: Restart Docker

templates/daemon.json.j2

@@ -0,0 +1,9 @@
+{
+  "default-runtime": "{{ nvidia_docker_default_runtime }}",
+  "runtimes": {
+    "nvidia": {
+      "path": "nvidia-container-runtime",
+      "runtimeArgs": []
+    }
+  }
+}