pymc-devs/pymc
Only sample using a single thread when sampling multiple chains in parallel
Geschlossen
#5.453 geöffnet am 09.02.2022
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.