Diffusion-MU-Attack icon indicating copy to clipboard operation
Diffusion-MU-Attack copied to clipboard

About multiple-gpus

Open CharlesGong12 opened this issue 1 year ago • 1 comments

Hi, do you remember me. Sorry to bother you again. I am a beginner of stable diffusion model and have some problems when running your code. Did you use a single gpu? I encountered CUDA out of memory with a single 3090-24GB and I could not fit the code into 2 or more gpus due to a lack of multiple-gpus' experience. Thank you very much for answering my rookie questions both this and last time.

CharlesGong12 avatar Nov 14 '23 16:11 CharlesGong12

No worries, our experiments were primarily conducted on a single NVIDIA V100 32 GB GPU. At the same time, we are actively working on adapting our codebase to support multi-GPU environments.

jinghanjia avatar Nov 19 '23 20:11 jinghanjia