diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard
Published 3 months ago β€’ verified publisher icon huggingface

A bug in multiple gpu training

Open kangxi1818 opened this issue 3 years ago β€’ 0 comments

Describe the bug

when train this code with multiple gpu ζˆͺ屏2023-02-20 δΈ‹εˆ9 37 27 I got a bug like this ζˆͺ屏2023-02-20 δΈ‹εˆ9 38 48

Reproduction

I just mutify this code ζˆͺ屏2023-02-20 δΈ‹εˆ9 40 14 change num_processes=1 to num_processes=2 before this bug another bug is like that RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method I change the 'fork' to 'spawn' like that ζˆͺ屏2023-02-17 δΈ‹εˆ4 37 21

Logs

No response

System Info

ζˆͺ屏2023-02-20 δΈ‹εˆ9 50 21 ζˆͺ屏2023-02-20 δΈ‹εˆ9 50 39 platform in jupyter notebook

kangxi1818 avatar Feb 20 '23 13:02 kangxi1818