Question

我经常在parallel::detectCores()中使用R来获取主机上的CPU核心数，以进行并行计算。我想计算一下计算中可用和空闲核心的数量。如果其他用户正在使用某些核心，我不想将它们用于我的程序。我们有办法以编程方式执行此操作吗？

Answer 1

这是一种使用系统命令和正则表达式来获取每个处理器的空闲时间的方法......这可能应该扩展到具有允许其他性能指标（即系统时间）的选项的函数。

library(doParallel)

# total cores
N_CORES <- detectCores()

# create list for readable lapply output
cores <- lapply(1:N_CORES, function(x) x - 1)
names(cores) <- paste0('CPU', 1:N_CORES - 1)

# use platform specific system commands to get idle time
proc_idle_time <- lapply(cores, function(x) {
  if (.Platform$OS.type == 'windows') {
    out <- system2(
      command = 'typeperf', 
      args = c('-sc', 1, sprintf('"\\processor(%s)\\%% idle time"', x)),
      stdout = TRUE)
    idle_time <- strsplit(out[3], ',')[[1]][2]
    idle_time <- as.numeric(gsub('[^0-9.]', '', idle_time))
  } else {
    # assumes linux
    out <- system2(
      command = 'mpstat', 
      args = c('-P', x),
      stdout = TRUE)
    idle_time <- as.numeric(unlist(strsplit(out[4], ' {2,}'))[12])
  }
  idle_time
})

检测R中的可用和空闲核心

1 个答案: