From 1eda0736a8f72076eca51faf543b3a825359b96e Mon Sep 17 00:00:00 2001
From: "Suren A. Chilingaryan" <csa@suren.me>
Date: Sat, 22 Feb 2020 01:30:41 +0100
Subject: Enable DKMS if CUDA is installed and disable cgroups in
 nvidia-container-runtime

---
 roles/cuda/tasks/cuda_init.yml | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'roles/cuda/tasks/cuda_init.yml')

diff --git a/roles/cuda/tasks/cuda_init.yml b/roles/cuda/tasks/cuda_init.yml
index ff54994..6d579de 100644
--- a/roles/cuda/tasks/cuda_init.yml
+++ b/roles/cuda/tasks/cuda_init.yml
@@ -31,6 +31,12 @@
     enabled: yes
   when: ansible_service_mgr == "systemd"
 
+- name: enable the DKMS systemd service
+  service:
+    name: dkms
+    enabled: yes
+  when: ansible_service_mgr == "systemd"
+
 - name: check if cuda_gpu_name0 ( /dev/nvidia0 ) exists
   stat:
     path: "{{ cuda_gpu_name0 }}"
-- 
cgit v1.2.3