如何使用GPU加速Pymc3采样？

Question

1. 我用'njobs'参数得到了多样本的结果，和我的预期相差甚远

2. 我更改了“.theanorc”文件以设置 'floatX'、'cnmem' 值等

3. 我已经通过命令'nvidia-smi'监控了GPU源码，很好用

但是，采样速度已经很慢了，比CPU还要慢。
这正常吗？

Answer 1

1. 这听起来像是收敛或模型构建的问题，与njobs或并行性无关。没有模型和痕迹，这里就没有太多可说的了。

GPU 仍处于实验阶段，我们已经看到某些模型 speed-ups 和其他模型 slow-downs。不过，ADVI 在 GPU 上似乎更容易 运行。您还可以检查所有模型类型和输入数据是否为 ​​float32。