如何通过 PYMC3 从 GPU 中受益

Question

我发现使用 GPU 与 CPU 时 PYMC3 速度的差异为零。

我正在拟合一个需要 500K+ 样本才能收敛的模型。显然它非常慢，所以我尝试使用 GPU 来加快速度（在 EC2 上使用 GPU 实例）。Theano 报告正在使用 GPU，因此我相信 CUDA/Theano 配置正确。然而，我强烈怀疑 Pymc3 没有利用 GPU。

• 我是否需要显式将变量设置为 TensorType(float32, scalar) ？目前，它们是 float64。
• 是否只有部分采样器/可能性可以从 CUDA 中受益？我正在拟合基于泊松的模型，因此使用 Metropolis 采样器，而不是 NUTS
• 有没有办法检查 pymc3 是否正在使用 GPU？