设置CPU核心数

Question

我在R下测试keras tensorflow，我想知道如何指定CPU内核的数量。我只在python下找到了没有keras的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()}