Skip to content

Instantly share code, notes, and snippets.

@ckrapu

ckrapu/pymc3-gpu-test.py

Last active November 21, 2021 20:59
Show Gist options
  • Save ckrapu/aa805de4afd23bc302c89a59f2f5382f to your computer and use it in GitHub Desktop.
Save ckrapu/aa805de4afd23bc302c89a59f2f5382f to your computer and use it in GitHub Desktop.
Download ZIP
PyMC3 GPU test
Raw
pymc3-gpu-test.py
import os
os.environ['THEANO_FLAGS'] = 'device=cuda,floatX=float64'
import pymc3 as pm
with pm.Model():
x = pm.Normal('x', shape=1000)
trace = pm.sample(chains=1, cores=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment