pymc-devs/pymc

Only sample using a single thread when sampling multiple chains in parallel

Geschlossen

#5.453 geöffnet am 09.02.2022

 (1 Kommentar) (0 Reaktionen) (0 zugewiesene Personen)Python (1.902 Forks)batch import
help wantedrequest discussionv4

Repository-Metriken

Stars
 (7.926 Sterne)
PR-Merge-Metriken
 (Durchschn. Merge 11T) (7 gemergte PRs in 30 T)

Beschreibung

My prior of what cores=4 does when calling pm.sample() is that pymc would use 4 of my cores. However, because of multi-threading this might not be the case. Furthermore, multi-threading is causing problems when running multiple chains in parallel: https://github.com/pymc-devs/pymc/issues/3140

I thus think we should set os.environ['OMP_NUM_THREADS'] = '1' if we're sampling on > 1 core.

Contributor Guide