Add driver version.

wookayin · Aug 8, 2024 · f1abc19 · f1abc19
1 parent dfce699
commit f1abc19
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/gpustat/rocml.py b/gpustat/rocml.py
@@ -82,7 +82,10 @@ def nvmlDeviceGetTemperature(handle, loc=NVML_TEMPERATURE_GPU):
 
 
 def nvmlSystemGetDriverVersion():
-    return ""
+    retval = rocmi.get_driver_version()
+    if retval is None:
+        return ""
+    return retval
 
 
 def check_driver_nvml_version(driver_version_str: str):

diff --git a/setup.py b/setup.py
@@ -91,7 +91,7 @@ def run(self):
 
 install_requires = [
     'nvidia-ml-py>=12.535.108',  # see #107, #143, #161
-    'rocmi>=0.2', # see #137
+    'rocmi>=0.3', # see #137
     'psutil>=5.6.0',    # GH-1447
     'blessed>=1.17.1',  # GH-126
     'typing_extensions',