training with multi-gpu

google / deepvariant

DeepVariant is an analysis pipeline that uses a deep neural network to call genetic variants from next-generation DNA sequencing data.

BSD 3-Clause "New" or "Revised" License

3.12k stars 702 forks source link

training with multi-gpu #832

Closed gambalab closed 1 week ago

gambalab commented 2 weeks ago

Hi, I was wondering if training with multiple GPUs is supported and, if so, if there is any specific flag to use or if it will be automatically activated when there is more than one GPU in your working environment.

Thank you in advance, Regards

akolesnikov commented 1 week ago

Hi @gambalab,

Unfortunately, DeepVariant does not support multi-GPU training (or inference) at this time.

gambalab commented 1 week ago

Thank you for the answer :)