CamDavidsonPilon / Probabilistic-Programming-and-Bayesian-Methods-for-Hackers

aka "Bayesian Methods for Hackers": An introduction to Bayesian methods + probabilistic programming with a computation/understanding-first, mathematics-second point of view. All in pure Python ;)
http://camdavidsonpilon.github.io/Probabilistic-Programming-and-Bayesian-Methods-for-Hackers/
MIT License
26.51k stars 7.84k forks source link

GPU/Graph for MCMC sampling on Tensorflow #452

Open bluesky314 opened 5 years ago

bluesky314 commented 5 years ago

I am using google colab. How to you activate GPU usage for MCMC sampling? I am finding it very slow.

And as building graphs is supposed to be faster than doing in eager exectuion. How do we convert the code of Ch2 example to graph? Can someone please share the code? Would appreciate any tips by the TF team why this is working so slow...