diff --git a/system/system_monitor/cmake/FindNVML.cmake b/system/system_monitor/cmake/FindNVML.cmake index 81ba07afd316b..0a79009ab98d2 100644 --- a/system/system_monitor/cmake/FindNVML.cmake +++ b/system/system_monitor/cmake/FindNVML.cmake @@ -20,24 +20,11 @@ # NVML_FOUND - True if NVML found. if(NOT NVML_INCLUDE_DIRS) - find_path(NVML_INCLUDE_DIRS nvml.h - PATHS - /usr/local/cuda/include - /usr/include - /usr/local/include - ) + find_path(NVML_INCLUDE_DIRS nvml.h PATHS /usr/local/cuda/include) endif() -# NVML library if(NOT NVML_LIBRARIES) - find_library(NVML_LIBRARIES NAMES nvidia-ml - PATHS - /usr/lib - /usr/lib64 - /usr/lib/x86_64-linux-gnu - /usr/local/lib - /usr/local/lib64 - ) + find_library(NVML_LIBRARIES NAMES nvidia-ml) endif() include(FindPackageHandleStandardArgs)