khanld / ASR-Wav2vec-Finetune

:zap: Finetune Wa2vec 2.0 For Speech Recognition
111 stars 23 forks source link

How to set up multi gpu training for this? #8

Closed ghosthunterk closed 1 year ago

ghosthunterk commented 1 year ago

I have 8 GPUs on my Lab server and the training only takes place on GPU 0, how can I set it up for it to train on multiple GPUs?

khanld commented 1 year ago

You can set device_ids=“0,1,2,3,4,5,6,7” to use all 8 gpus