nvidia: use udev rule for the symlinks

saltyorg · Mar 2, 2024 · 926f0bd · 926f0bd
1 parent fba24b2
commit 926f0bd
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 5 deletions.
diff --git a/roles/docker/templates/nvidia-persistenced.service.j2 b/roles/docker/templates/nvidia-persistenced.service.j2
@@ -4,9 +4,6 @@ Wants=syslog.target
 
 [Service]
 Type=forking
-{% if nvidia_docker_enable_nvidia_ctk %}
-ExecStartPre=/usr/bin/nvidia-ctk system create-dev-char-symlinks --create-all
-{% endif %}
 ExecStart=/usr/bin/nvidia-persistenced
 ExecStopPost=/bin/rm -rf /var/run/nvidia-persistenced
 

diff --git a/roles/nvidia/defaults/main.yml b/roles/nvidia/defaults/main.yml
@@ -51,8 +51,6 @@ 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/99-nvidia-udev.rules b/roles/nvidia/files/99-nvidia-udev.rules
@@ -0,0 +1,2 @@
+# This will create /dev/char symlinks to all device nodes
+ACTION=="add", DEVPATH=="/bus/pci/drivers/nvidia", RUN+="/usr/bin/nvidia-ctk system create-dev-char-symlinks --create-all"
diff --git a/roles/nvidia/tasks/main.yml b/roles/nvidia/tasks/main.yml
@@ -131,6 +131,19 @@
           ansible.builtin.include_tasks: "subtasks/patch.yml"
           when: (not nvidia_patch_backup_folder.stat.exists)
 
+    # Install Nvidia udev rule
+
+    - name: Copy the NVIDIA udev rule file
+      ansible.builtin.copy:
+        src: 99-nvidia-udev.rules
+        dest: /etc/udev/rules.d/99-nvidia-udev.rules
+        owner: root
+        group: root
+        mode: '0644'
+
+    - name: Reload udev rules
+      ansible.builtin.shell: udevadm control --reload-rules && udevadm trigger
+
     # Install Nvidia Runtime Container
 
     - name: "Nvidia Docker Task"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		# This will create /dev/char symlinks to all device nodes
		ACTION=="add", DEVPATH=="/bus/pci/drivers/nvidia", RUN+="/usr/bin/nvidia-ctk system create-dev-char-symlinks --create-all"