pymc3:njobs>1 与 GPU 的并行计算

Question

我正在尝试通过并行化加速 pymc3 采样，但我只看到了适度的好处。

我能够在 i7 MacBook Pro 上将总运行时间从 25 分钟(njobs=1)减少到 13 分钟(njobs=6)。由于pymc实际开始采样需要4分钟左右的时间，所以增加的幅度比较小。

问题是 - 有没有人成功地将 GPU 与 pymc3 一起使用？对于需要 6-8 分钟采样的模型，我可以获得多少好处？ (我的 MacBook 有 nvidia GT 750M 2Gb)

Answer 1

我在 Intel i7-4930 上运行 Linux。

我运行了一个 PyMC3 模型，它在 CPU 上运行了 90 分钟(利用了所有内核)，但在我的 GeForce GTX 970 上只运行了 18 分钟。

因此，速度提高了近 5 倍。