python-3.x - 升级到更强大的新笔记本电脑后，pymc3 运行非常缓慢

Question

import pymc3 as pm 
import numpy as np 

x = np.linspace(0,1,100)
y_true = 3*x + 5 
y_obs = y_true + np.random.normal(loc=0, scale=0.02,size=100)

with pm.Model() as model:
    a = pm.Normal('a', mu=2.0, sigma=3.0) 
    b = pm.Normal('b', mu=2.0, sigma=3.0) 
    y_model = a*x + b 
    s = pm.HalfNormal('s', sigma=0.05)
    likelihood = pm.Normal('y', mu=y_model, sigma=s, observed = y_obs)

trace = pm.sample(500, tune=500, chains=2, cores=1, model=model)

我升级到新的笔记本电脑，pymc3 运行非常缓慢。在我的旧笔记本电脑上需要 2-3 秒的简单线性回归现在需要大约 10 分钟！

笔记本电脑：Windows 10，Intel(R) Core(TM) i9-9880H CPU @ 2.30GHz 2.30 GHz，32 GB RAM。

python: 3.7 pymc3: 3.9.3 Theano: 1.0.5 我没有安装 m2w64-toolchain。试过了，pymc3 甚至没有加载。所以，在那之后重新安装了Anaconda。

有什么想法可以让它这么慢吗？这是一个theano问题吗？

谢谢苏曼思

标签： python-3.xtheanopymc3