AI学习:我用8GB显存尝试SDXL,结果让我大跌眼镜!
一、SDXL模型:内存溢出问题
最近我从Hugging Face下载了SDXL基础模型,并将其放在automatic1111的models/stablediffusion文件夹中。然而,当我尝试在512×512(批次为1)上运行该模型时,它首先完成了一次传递(20步),然后崩溃并显示内存溢出的错误。我试图将分辨率降低到256×256,但没有起到作用。它给出了以下错误信息:OutOfMemoryError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 7.92 GiB total capacity; 6.98 GiB already allocated; 39.44 MiB free; 7.38 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. 有人知道如何解决这个问题吗?我在Twitter上看到(https://twitter.com/EMostaque/status/1671211689633611776?cxt=HHwWgIC8kayLq7EuAAAA)说它应该在只有8GB内存的情况下仍然可以工作。
二、尝试解决内存溢出问题
这个问题真是让人头疼!我试图降低分辨率,但没有任何改善。我还尝试了一些其他方法,但都没有起作用。难道我做错了什么吗?难道我的电脑配置不够强大吗?这可真是让人沮丧!
三、寻求帮助
我决定向社区求助。也许有其他人遇到过类似的问题并找到了解决方法。如果你知道如何解决这个内存溢出问题,请在下方留言告诉我。我非常感激!
总结
SDXL模型在我的电脑上出现了内存溢出的问题,我尝试了降低分辨率等方法,但都没有解决。如果你有任何解决方法,请务必告诉我。非常感谢你的阅读和帮助!
感谢你的阅读!