MultiVi with multiple GPUs and CPUs

Dear scvi-tools developers and community,

I was wondering if it is possible to run MultiVi with multiple GPUs and CPUs.
In case it is, how can I set up the number of GPUs and CPUs for use of MultiVi?
Thank you for your attention.

With best wishes,

Daniele Muraro

You should be able to set the number of threads here:

https://docs.scvi-tools.org/en/stable/api/reference/scvi._settings.ScviConfig.html#scvi._settings.ScviConfig

But we only currently support single gpu training. We are exploring options for multi gpu training

I realised that I didn’t reply, I apologise for that. Thank you so much for letting me know :slight_smile: