multiVI speed and cpu consumption

Help
1

Hi developers, I am using multiVI (from scvi-tools 1.3.2) to analyze the processed pbmc10k data following the tutorial. For some reason, training is really slow, it takes 15 seconds for 1 epoch. In addition, multiVI is consuming all 80 cpus on my server despite having set

scvi.settings.num_threads = 1

Is there something else i can do to limit the number of CPUs multiVI uses and speed up training?

2

I’m technically not a developer, but can you share why you are training on CPUs?
I’m surprised it only took 15 seconds for 1 epoch. That’s a great speed for CPU training.
Only way to speed that up is the down-sample your data, or move to GPUs.

3

i see, thank you so much for your response!