nvidia: add nvidia-ctk toggle

saltyorg · Feb 18, 2024 · 1783d43 · 1783d43
1 parent c596207
commit 1783d43
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/roles/docker/files/nvidia-persistenced.service b/roles/docker/files/nvidia-persistenced.service
@@ -4,8 +4,10 @@ Wants=syslog.target
 
 [Service]
 Type=forking
+{% if nvidia_docker_enable_nvidia_ctk %}
+ExecStartPre=/usr/bin/nvidia-ctk system create-dev-char-symlinks --create-all
+{% else %}
 ExecStart=/usr/bin/nvidia-persistenced
-ExecStartPre=/usr/bin/nvidia-ctk system create-dev-char-symlinks --create-all --load-kernel-modules --create-device-nodes
 ExecStopPost=/bin/rm -rf /var/run/nvidia-persistenced
 
 [Install]

diff --git a/roles/nvidia/defaults/main.yml b/roles/nvidia/defaults/main.yml
@@ -51,6 +51,8 @@ nvidia_docker_runtime_docker_daemon_json_jq_command: |
       | .runtimes.nvidia.runtimeArgs = []' \
       /etc/docker/daemon.json | jq --sort-keys | sponge /etc/docker/daemon.json
 
+nvidia_docker_enable_nvidia_ctk: true
+
 ################################
 # Kernel
 ################################

diff --git a/roles/nvidia/files/nvidia-persistenced.service b/roles/nvidia/files/nvidia-persistenced.service
@@ -4,8 +4,10 @@ Wants=syslog.target
 
 [Service]
 Type=forking
-ExecStart=/usr/bin/nvidia-persistenced
+{% if nvidia_docker_enable_nvidia_ctk %}
 ExecStartPre=/usr/bin/nvidia-ctk system create-dev-char-symlinks --create-all
+{% else %}
+ExecStart=/usr/bin/nvidia-persistenced
 ExecStopPost=/bin/rm -rf /var/run/nvidia-persistenced
 
 [Install]

diff --git a/roles/nvidia/tasks/subtasks/docker.yml b/roles/nvidia/tasks/subtasks/docker.yml
@@ -51,6 +51,7 @@
 
 - name: Docker | Create '/dev/char' symlinks
   ansible.builtin.shell: nvidia-ctk system create-dev-char-symlinks --create-all
+  when: nvidia_docker_enable_nvidia_ctk
 
 - name: Docker | Populate Service Facts
   ansible.builtin.service_facts: