neonbjb / DL-Art-School

DLAS - A configuration-driven trainer for generative models
Apache License 2.0
137 stars 135 forks source link

Multi GPU Training #13

Closed capmolyneux closed 1 year ago

capmolyneux commented 1 year ago

Multi GPU Memory Issue- I select both GPU indices in training set up, but I am getting memory errors when loading up my GPUs, after the first is exhausted it throws an OOM error before loading up second GPU