[nvidia] Capture more nvidia commands

Capture commands related to nvidia container toolkit. Related: RHEL-58172 Signed-off-by: Jose Castillo <[email protected]>
sosreport · Oct 1, 2024 · 7296934 · 7296934
1 parent 96e4d79
commit 7296934
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/sos/report/plugins/nvidia.py b/sos/report/plugins/nvidia.py
@@ -16,7 +16,9 @@ class Nvidia(Plugin, IndependentPlugin):
 
     short_desc = 'Nvidia GPU information'
     plugin_name = 'nvidia'
-    commands = ('nvidia-smi',)
+    commands = ('nvidia-smi', 'nvidia-ctk',)
+    services = ('nvidia-persistenced', 'nvidia-fabricmanager',
+                'nvidia-toolkit-firstboot')
 
     def setup(self):
         subcmds = [
@@ -30,8 +32,15 @@ def setup(self):
             'nvlink -e'
         ]
 
-        self.add_service_status("nvidia-persistenced")
+        ctk_subcmds = [
+            'cdi list',
+            '--version',
+        ]
+
+        self.add_copy_spec("/etc/cdi/nvidia.yaml")
+
         self.add_cmd_output([f"nvidia-smi {cmd}" for cmd in subcmds])
+        self.add_cmd_output([f"nvidia-ctk {cmd}" for cmd in ctk_subcmds])
 
         query = ('gpu_name,gpu_bus_id,vbios_version,temperature.gpu,'
                  'utilization.gpu,memory.total,memory.free,memory.used,'
@@ -44,4 +53,5 @@ def setup(self):
         )
         self.add_journal(boot=0, identifier='nvidia-persistenced')
 
+
 # vim: set et ts=4 sw=4 :