scala - Nd4j：使用多线程比单线程慢

Question

这是我的处理器：MacOs Sierra 上的 2.3 GHz Intel Core i7（即 4 个超线程内核）

这是我的程序：

package nn

import org.nd4j.linalg.api.ndarray.INDArray
import org.nd4j.linalg.factory.Nd4j.randn
import org.nd4j.linalg.ops.transforms.Transforms._
import org.nd4s.Implicits._

object PerfTest extends App {

  val topology = List(784, 30, 10)
  val biases: List[INDArray] =
    topology.tail.map(size => randn(size, 1))

  val weights: List[INDArray] =
    topology.sliding(2).map(t => randn(t(1), t.head)) toList

  (1 to 100000).foreach { i =>
    val x = randn(784, 1)
    biases.zip(weights).foldLeft(List(x)) {
      case (as, (b, w)) =>
        val z = (w dot as.last) + b
        val a = sigmoid(z)
        as :+ a
    }
  }
}

当我使用默认线程运行上述程序时（对于 nd4j 和此处理器，这将是 4），大约需要 28 秒。

当我在 1 个核心 ( export OMP_NUM_THREADS=1) 上运行它时，需要 18 秒。

知道这是为什么吗？谢谢你。

Answer 1

我无法找到对此的澄清。所以我毫不费力地迁移到了 Breeze，它的速度快了 6 倍。