I’ve run an MCMC sampling inference on a model I have on the GPU. The sampling finished successful and I was able to get posterior samples.
However when trying to vizualize my samples and traces using Arviz
az.plot_trace I get a
RuntimeError: RESOURCE_EXHAUSTED: Out of memory while trying to allocate 11044800000 bytes error. I have been trying to move the
numpyro.infer.mcmc.MCMC object resulting from the sampling to the CPU to avoid this error. However, I haven’t been able to figure this out how to do this.
Is there any way I could move the
numpyro.infer.mcmc.MCMC object resulting from the sampling from GPU to CPU?