Update zeus/device/gpu/amd.py

Co-authored-by: Jae-Won Chung <[email protected]>
ml-energy · Oct 29, 2024 · 6e632d2 · 6e632d2
1 parent 8405392
commit 6e632d2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/zeus/device/gpu/amd.py b/zeus/device/gpu/amd.py
@@ -51,7 +51,10 @@ def amdsmi_is_available() -> bool:
         return False
     # usually thrown if versions of amdsmi and ROCm are incompatible.
     except AttributeError:
-        logger.warning("Do you have the correct version of ROCm and amdsmi installed?")
+        logger.warning(
+            "Failed to import amdsmi. "
+            "Ensure amdsmi's version is at least as high as the current ROCm version."
+        )
         return False
     try:
         amdsmi.amdsmi_init()