본문 바로가기
잡담/WebUI(stable-diffusion)

WebUI CUDA out of memory 에러 해결방법

by flatsun 2023. 4. 9.
반응형

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 사용량 줄이는 방법

WebUI 에서 그래픽카드의 VRAM이 작아도 이미지를 생성하고 싶다면 시간과 퀄리티를 희생하는 대신 VRAM 사용량을 줄여줄 수 있는데 방법은 WebUI 설치 폴더로 이동한 뒤 webui-user.bat 파일 우클릭 후

flatsun.tistory.com

 

반응형

댓글