Cobaya multiple cpus per chain
Posted: October 02 2019
Hi,
I was trying to set up my python script based on Cobaya to perform a parallel mcmc sampling for multiple chains. I would like to have several cores per chain.
As I am not sure if I did things correctly, to speed up sampling, I am posting here: (or maybe I could simply use the flag time for a simple script, but I put the command here as it can be useful for someone else if correct).
On Nersc-Cori, I use
srun -n-tasks=8 --cpus-per-task=8 python cobaya_test.py
-n-tasks will specify the number of chains
--cpus-per-task cpus per chain
Does this make sense?
Best,
Omar
I was trying to set up my python script based on Cobaya to perform a parallel mcmc sampling for multiple chains. I would like to have several cores per chain.
As I am not sure if I did things correctly, to speed up sampling, I am posting here: (or maybe I could simply use the flag time for a simple script, but I put the command here as it can be useful for someone else if correct).
On Nersc-Cori, I use
srun -n-tasks=8 --cpus-per-task=8 python cobaya_test.py
-n-tasks will specify the number of chains
--cpus-per-task cpus per chain
Does this make sense?
Best,
Omar