Hello pmyc3 experts I’m trying to run pymc3 with GPU support with following setup: ubuntu-20.04, kernel 5.4.0-72-generic I created a conda environment and installed: pip install pymc-3.11.0 conda install -c conda-forge pygpu I installed cuda and cudnn from the nvidia site. I can run nvidida-smi…

Pymc3-3.11.0 with GPU support

twiecki June 23, 2021, 7:43am 9

Try setting

# Disable C compilation by default
theano.config.cxx = ""
# This will make the JAX Linker the default
theano.config.mode = "JAX"

And then just running ADVI as usual.

With all the changes, what are the options for ADVI training on GPUs?

Topic		Replies	Views
Setup environment for using GPU with Theano Questions gpu	1	3395	July 24, 2018
"Your cuDNN version is more recent than Theano" Questions	1	1076	December 20, 2021
Leveraging the GPU in PyMC3 Questions gpu , theano	5	8010	August 12, 2017
Using Multiple GPU's Questions gpu , theano	0	710	August 28, 2018
Docker GPU image for PyMC3 Questions development_env	4	895	December 10, 2021