justinpinkney / stable-diffusion

MIT License
1.45k stars 266 forks source link

First GPU use more memory,and lead to OOM #79

Open Lzhang-hub opened 1 year ago

Lzhang-hub commented 1 year ago

Condition:

GPU: A100 40G *8 batch size=2

error

CUDA out of memory.

some confusion

We find the gpu 0 use more memory than other gpus image

And there are 7 additional processes on GPU 0. image

What are these 7 processes used for? Or Is there any way to reduce the memory usage of GPU 0?