...

ansible/roles/ollama/defaults/main.yml

@@ -0,0 +1,5 @@
+---
+ollama_install_dir: /opt/ollama
+ollama_image: ollama/ollama
+ollama_image_tag: latest
+ollama_container_labels: []

ansible/roles/ollama/tasks/deploy/main.yml

@@ -0,0 +1,18 @@
+---
+- name: Create Ollama Directories
+  ansible.builtin.file:
+    state: directory
+    path: "{{ ollama_install_dir }}/{{ item }}"
+  loop:
+  - data
+  - compose
+
+- name: Pull Ollama Image
+  community.docker.docker_image:
+    source: pull
+    name: "{{ ollama_image }}:{{ ollama_image_tag }}"
+
+- name: Copy composefile
+  ansible.builtin.template:
+    src: templates/composefile.yml
+    dest: "{{ ollama_install_dir }}/compose/composefile.yml"

ansible/roles/ollama/tasks/main.yml

@@ -0,0 +1,5 @@
+---
+- name: Include selected operations for Ollama
+  tags:
+  - ollama
+  ansible.builtin.include_tasks: "{{ deployment_action|default('deploy') }}/main.yml"

ansible/roles/ollama/templates/composefile.yml

@@ -0,0 +1,20 @@
+services:
+
+  ollama:
+    image: {{ ollama_image }}:{{ ollama_image_tag }}
+    deploy:
+      resources:
+        reservations:
+          devices:
+          - driver: nvidia
+            count: all
+            capabilities: [gpu]
+    network_mode: host
+    volumes:
+    - {{ ollama_install_dir }}/data:/root/.ollama
+    tty: true
+    restart: unless-stopped
+    {% if ollama_container_labels %}
+    labels:
+    {{ ollama_container_labels|to_nice_yaml|indent(4) }}
+    {% endif %}