Skip to content

Commit 881c903

Browse files
List system CTK libraries
1 parent dc4e500 commit 881c903

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

rapids_cli/debug/debug.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import sys
1010
from datetime import datetime
1111
from importlib.metadata import distributions, version
12+
from pathlib import Path
1213

1314
import pynvml
1415
from 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+
8793
def 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(),

0 commit comments

Comments
 (0)