Check /sys/module/amdgpu for ROCk instead of lsmod

Closes: #102 Signed-off-by: YiyangWu <xgreenlandforwyy@gmail.com>
ROCm · Apr 2, 2022 · dc2a4b5 · dc2a4b5
1 parent bd95425
commit dc2a4b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python_smi_tools/rocm_smi.py b/python_smi_tools/rocm_smi.py
@@ -66,7 +66,7 @@ def driverInitialized():
     """
     driverInitialized = ''
     try:
-        driverInitialized = str(subprocess.check_output("cat /proc/modules|grep amdgpu", shell=True))
+        driverInitialized = str(subprocess.check_output("test -d /sys/module/amdgpu", shell=True))
     except subprocess.CalledProcessError:
         pass
     if len(driverInitialized) > 0: