Use relative path to find librocm_smi

Change-Id: Ifca3f54d680a802c1c5fa360d17e64338b9ac9a8
ROCm · Oct 28, 2020 · 0ee670d · 0ee670d
1 parent a45d958
commit 0ee670d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python_smi_tools/rsmiBindings.py b/python_smi_tools/rsmiBindings.py
@@ -7,7 +7,10 @@
 from ctypes import *
 from enum import Enum
 
-path_librocm = 'librocm_smi64.so'
+import os
+
+# Use ROCm installation path if running from standard installation
+path_librocm = os.path.dirname(os.path.realpath(__file__)) + '/../lib/librocm_smi64.so'
 # ----------> TODO: Support static libs as well as SO
 
 try: