File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed
Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change 99import sys
1010from datetime import datetime
1111from importlib .metadata import distributions , version
12+ from pathlib import Path
1213
1314import pynvml
1415from rich .console import Console
@@ -84,6 +85,11 @@ def gather_conda_packages():
8485 return "Conda not installed"
8586
8687
88+ def gather_system_ctk ():
89+ """Return system ctk."""
90+ return sorted ([str (p ) for p in Path ("/usr/local" ).glob ("cuda*" ) if p .is_dir ()])
91+
92+
8793def gather_package_managers ():
8894 """Return package managers."""
8995 return {
@@ -102,6 +108,7 @@ def run_debug(output_format="console"):
102108 "nvidia_smi_output" : gather_nvidia_smi_output (),
103109 "driver_version" : gather_driver_version (),
104110 "cuda_version" : gather_cuda_version (),
111+ "system_ctk" : gather_system_ctk (),
105112 "python_version_full" : gather_python_version_full (),
106113 "python_version" : gather_python_version (),
107114 "package_versions" : gather_package_versions (),
You can’t perform that action at this time.
0 commit comments