Out of memory in ubuntu 22.04 when use gradio_app_mv.py, but can be used in windows in 8G GPU

xxlong0 / Wonder3D

Single Image to 3D using Cross-Domain Diffusion for 3D Generation

https://www.xxlong.site/Wonder3D/

GNU Affero General Public License v3.0

4.49k stars 351 forks source link

Out of memory in ubuntu 22.04 when use gradio_app_mv.py, but can be used in windows in 8G GPU #153

Open hackt0 opened 3 months ago

hackt0 commented 3 months ago

I can use by run_test.sh, but when I use gradio_app_mv.py to start web UI. I will get error: torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU 0; 7.79 GiB total capacity; 7.03 GiB already allocated; 384.69 MiB free; 7.13 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. but in windows it works fine.