How to utilize all GPUs?

fastai
1

I’m training an image segmentation model using fasai ‘unet_learner.’ While tracking the model progress via wandb I noticed that my code is only using a single GPU. How can I make sure I use all GPUs?

image
image800×339 16.6 KB

1 Like
2

You need to use multi-GPU distributed training:

1 Like