From e6cbaa5fc2688dd3de004c140eaba43220b4a2fe Mon Sep 17 00:00:00 2001
From: Nicole O'Connor <nicole@otl-hga.net>
Date: Wed, 31 Jul 2024 13:21:39 -0700
Subject: [PATCH] docker/nvidia

---
 deploy-docker.yaml | 42 ++++++++++++++++++++++++++++++++++++++++++
 deploy-nvidia.yaml | 17 +++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 deploy-docker.yaml
 create mode 100644 deploy-nvidia.yaml

diff --git a/deploy-docker.yaml b/deploy-docker.yaml
new file mode 100644
index 0000000..688f6a0
--- /dev/null
+++ b/deploy-docker.yaml
@@ -0,0 +1,42 @@
+---
+- name: "Setup Docker"
+  become: true
+  tasks:
+   - name: "Download Docker Repository Key"
+     # HACK: apt_key has been deprecated, and its current (2023) replacement appears to be nothing
+     #ansible.builtin.apt_key:
+     # url: https://download.docker.com/linux/ubuntu/gpg
+     # id: 7EA0A9C3F273FCD8
+     # state: present
+     ansible.builtin.get_url:
+      url: https://download.docker.com/linux/ubuntu/gpg
+      dest: /tmp/docker.gpg.armor
+   - name: "Add Docker Repository Key"
+     ansible.builtin.shell:
+      cmd: gpg --dearmor -o /etc/apt/keyrings/docker.gpg < /tmp/docker.gpg.armor
+      creates: /etc/apt/keyrings/docker.gpg
+   - name: "Add Docker Repository"
+     ansible.builtin.apt_repository:
+      repo: "deb [arch=amd64 signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu {{ ansible_distribution_release }} stable"
+      filename: docker
+      state: present
+   - name: "Install Docker packages"
+     ansible.builtin.apt:
+      name:
+       - docker-ce
+       - docker-ce-cli
+       - docker-buildx-plugin
+       - docker-compose-plugin
+       - python3-docker
+       - unzip
+   - name: "Add self to docker group"
+     ansible.builtin.user:
+      name: "{{ ansible_user }}"
+      append: true
+      groups:
+       - docker
+   - name: "Start Docker"
+     ansible.builtin.service:
+      name: docker
+      state: started
+      enabled: true
\ No newline at end of file
diff --git a/deploy-nvidia.yaml b/deploy-nvidia.yaml
new file mode 100644
index 0000000..0b9ed04
--- /dev/null
+++ b/deploy-nvidia.yaml
@@ -0,0 +1,17 @@
+---
+- name: Setup NVIDIA drivers
+  become: true
+  tasks:
+   - name: "Adding NVIDIA CUDA repository"
+     ansible.builtin.apt_repository:
+      repo: "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
+      filename: nvidia-upstream-ppa
+   - name: "Installing CUDA"
+     ansible.builtin.apt:
+      pkg:
+       - cuda-12
+       - nvidia-headless
+   - name: "Making sure NVIDIA UVM is working"
+     community.general.modprobe:
+      name: "nvidia_uvm"
+      state: present
\ No newline at end of file