Leveraging the GPU in PyMC3

Hi Thomas and Adrian,

So updating to the latest master and removing njobs=2 did the trick! It sounds like Theano will take advantage of multiple GPUs then? I’m currently running a p2.xlarge instance but can easily switch that to a p2.8xlarge.

Appreciate your quick response! Definitely looking forward to diving into PyMC3.

Have a great weekend,

Chris

1 Like