dreambooth icon indicating copy to clipboard operation
dreambooth copied to clipboard

Model doesn't appear to be using all available GPUs

Open rdev opened this issue 2 years ago • 1 comments

I ran training on this model with Cog on a 1xA100 instance and 8xA100 instance and got essentially the same performance. Is the container utilizing all available GPUs?

cc @chenxwh

rdev avatar Nov 15 '22 22:11 rdev

I don't think it implements multi-gpu training? So essentially just one gpu is used

chenxwh avatar Nov 15 '22 22:11 chenxwh