sail-sg / EditAnything

Edit anything in images powered by segment-anything, ControlNet, StableDiffusion, etc. (ACM MM)
Apache License 2.0
3.3k stars 188 forks source link

What is the requirement for GPU to deploy on my own serer? #13

Open Dinxin opened 1 year ago

Dinxin commented 1 year ago

I want to deploy this project on my server with one V100 GPU (32GB). However, I found when this webui launches, it occupies about 28GB GPU memory and cannot produce result on my server.

gasvn commented 1 year ago

We are using the A100 with 40G memory. You can use each demo instead of all demos at once. For example, you can run with python sam2edit.py. This requires less than 16G to run. Please let me know if you still cannot run it.

Dinxin commented 1 year ago

I only launched the sam2edit.py on a V100 GPU with 32GB. But I found my program blocked. image and the memory occupation is extensive. image What happened?

Dinxin commented 1 year ago

Seemingly, it does not employ the GPU, although it occupies the GPU memory. image image

gasvn commented 1 year ago

It seems your model uses the cpu instead of gpu. I suggest you try to reinstall the packages as listed in the https://github.com/sail-sg/EditAnything/blob/main/requirements.txt