fix: improve FindNVML.cmake

Signed-off-by: ito-san <[email protected]>
tier4 · Dec 16, 2024 · 57706f7 · 57706f7
1 parent 6eb5fd3
commit 57706f7
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/system/system_monitor/cmake/FindNVML.cmake b/system/system_monitor/cmake/FindNVML.cmake
@@ -20,11 +20,24 @@
 #  NVML_FOUND        - True if NVML found.
 
 if(NOT NVML_INCLUDE_DIRS)
-  find_path(NVML_INCLUDE_DIRS nvml.h PATHS /usr/local/cuda/include)
+  find_path(NVML_INCLUDE_DIRS nvml.h
+    PATHS
+      /usr/local/cuda/include
+      /usr/include
+      /usr/local/include
+  )
 endif()
 
+# NVML library
 if(NOT NVML_LIBRARIES)
-  find_library(NVML_LIBRARIES NAMES nvidia-ml)
+  find_library(NVML_LIBRARIES NAMES nvidia-ml
+    PATHS
+      /usr/lib
+      /usr/lib64
+      /usr/lib/x86_64-linux-gnu
+      /usr/local/lib
+      /usr/local/lib64
+  )
 endif()
 
 include(FindPackageHandleStandardArgs)