Amazon ec2 如何从使用PYMC3的GPU中获益

当使用GPU和CPU时，我看到PYMC3的速度没有任何差别

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

• 我是否需要显式地将变量设置为TensorType（float32，标量）？目前，它们是64
• 是否只有一些采样器/可能性可以从CUDA中受益？我正在拟合基于泊松的模型，所以使用Metropolis采样器，而不是NUTS
• 有没有办法检查pymc3是否使用GPU

---

我刚刚运行了a的示例，也找不到任何加速度。我想我们需要关于如何设置模型的信息，以便theano能够使用GPU（或多线程）。奇怪的是，是什么让你认为GPU没有被使用。您是否能够运行（例如）nvidia smi命令来检查使用情况？我在使用GPU加速方面取得了一些成功，但并没有解决大问题。听起来pymc3开发GPU还有很多工作要做，因为我们已经用nvidia smi进行了测试，在运行采样器时没有使用GPU。然而，Theano报告说正在使用GPU。Theano 0.8.0文档（）上说：透明使用GPU–执行数据密集型计算的速度比使用CPU快140倍。*（仅限浮点32）*。您是否尝试过将数据类型设置为float32，正如您在问题中所说的那样？