diff --git a/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py b/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py index 87f0bb951a..91b6751634 100644 --- a/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py +++ b/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py @@ -7,7 +7,10 @@ import ctypes.util 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: