diff --git a/core/src/server/init/GpuChecker.cpp b/core/src/server/init/GpuChecker.cpp index 4d3dba8850..dbd19d124a 100644 --- a/core/src/server/init/GpuChecker.cpp +++ b/core/src/server/init/GpuChecker.cpp @@ -188,11 +188,7 @@ GpuChecker::CheckGpuEnvironment() { char device_name[NVML_DEVICE_NAME_BUFFER_SIZE]; int major, minor; - for (uint32_t i = 0; i < device_count; i++) { - if (gpu_sets.find(i) == gpu_sets.end()) { - continue; - } - + for (uint32_t i = 0; i < gpu_sets.size(); i++) { nvmlDevice_t device; nvmlresult = nvmlDeviceGetHandleByIndex(i, &device); fiu_do_on("GpuChecker.CheckGpuEnvironment.nvml_get_device_handle_fail", nvmlresult = NVML_ERROR_UNKNOWN);