Ошибка нехватки памяти FLuxgym
Краткое содержание
Всем привет. У меня возникла ошибка (OOM CUDA) в Fluxgym. Я использую RTX3060 с 12 ГБ видеопамяти и 16 ГБ оперативной памяти, и получаю следующую ошибку: :([INFO] torch.OutOfMemoryError: Недостаточно памяти CUDA. Попытка выделить 54,00 MiB. У GPU 0 общая емкость составляет 12,00 GiB, из которых свободно 0 байт. Из выделенной памяти PyTorch выделил 11,50 GiB, а на 5,09 MiB зарезервировано, но не выделено. Если объем зарезервированной, но не выделенной памяти большой, попробуйте установить PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True, чтобы избежать фрагментации. См. документацию по управлению памятью (https://docs.pytorch.org/docs/stable/notes/cuda.html#optimizing-memory-usage-with-pytorch-cuda-alloc-conf)). У кого-нибудь есть решение? Опубликовано пользователем /u/Orvast[ссылка][комментарии]
Полный текст статьи пока не загружен.