设置CPU核心数

Question

我正在测试R下的keras tensorflow，我想知道如何指定CPU内核的数量。我只在没有keras的python下找到tensorflow解决方案。

> library(keras)
> library(doParallel)
> cl <- makeCluster(2)
> registerDoParallel(cl)
> is_keras_available()
[1]TRUE
> foreach(i=1:3) %dopar% { is_keras_available()}
Error in { : 
  task 1 failed - "impossible to find the function "is_keras_available""
> parallel::stopCluster(cl)

Answer 1

使用doParallel库，您可以使用detectCores()函数检测核心数，并指定要用于registerDoParallel()函数的核心数。

看看https://cran.r-project.org/web/packages/doParallel/vignettes/gettingstartedParallel.pdf

请务必通过.export参数

将您的库包含在代码的并行化中

foreach(i=1:3, .packages = c("keras")) %dopar% { is_keras_available()}