WebUI CUDA out of memory 에러 해결방법

OutOfMemoryError: CUDA out of memory. Tried to allocate 9.00 GiB (GPU 0; 23.99 GiB total capacity; 15.96 GiB already allocated; 5.02 GiB free; 16.30 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

 

WebUI 사용 시 위와 같이

OutOfMemoryError: CUDA out of memory 에러가 발생하면

이미지를 생성하다가 VRAM이 부족해서 발생한 것인데

 

해결방법은 Width / Height를 줄여주면

사용하는 VRAM 자체가 줄어든다

 

그리고 Batch size의 경우에도

한번에 여러 이미지를 그리는 것이기 떄문에

1로 변경해주면 도움이 된다

 

이렇게 해도 안된다면

--lowvram / --medvram

등의 옵션을 넣어 실행하는 것을 권장한다

WebUI VRAM 사용량 줄이는 방법