Hello,
I switched from a standard, CPU environment in GDC to one with a few GPUs. Now, I’m getting the following error.
RuntimeError: module compiled against API version 0xe but this version of numpy is 0xd ERROR (aesara.graph.opt): Optimization failure due to: constant_folding ERROR (aesara.graph.opt): node: InplaceDimShuffle{}(TensorConstant{(1,) of 356228}) ERROR (aesara.graph.opt): TRACEBACK: ERROR (aesara.graph.opt): Traceback (most recent call last):
This is the system I’m running:
PyMC Version: 4.0.0b6
Aesara Version: 2.5.1
Arvize Verions: 0.12.1
Here is the model in case that is needed:
with pm.Model(check_bounds=False) as linear:
a = pm.Normal("a", mu=0, sigma=.05)
b = pm.Normal("b", mu=0, sigma=.05)
error = pm.HalfNormal("error", sigma=.1)
trend = pm.Deterministic("trend", α + β * t)
pm.Normal("likelihood", mu=trend, sigma=error, observed=y)
linear_prior_predictive = pymc.sampling_jax.sample_numpyro_nuts(tune=1000, chains = 4)