Google Notebook运行Disco Diffusion时CUDA内存不足问题求助
解决Disco Diffusion生成光流图时的CUDA显存不足问题
这个错误就是GPU显存不够用了——当前你的GPU已经用了13.05GiB,只剩334MiB,但生成光流图需要508MiB,所以装不下报错。试试这几个简单方法,都是新手能快速上手的:
- 降低图像分辨率:找到Disco Diffusion设置里的「Width」(宽度)和「Height」(高度),把数值改小,比如从1024×512改成768×384,甚至640×320。分辨率越低,需要的显存越少,是最直接的解决办法。
- 改成单张生成:如果设置里有「Batch Size」(批量生成数),把它改成1,别一次生成多张图,先单张跑通再说。
- 重启运行环境:在Google Colab顶部点「Runtime」→「Restart Runtime」,重启后之前占用的显存会被清空,再重新运行代码。
- 添加显存优化代码:在代码最开头加上这两行,帮PyTorch更合理分配显存:
第一行是清空GPU缓存,第二行是调整显存块拆分大小,减少碎片化,挤出更多可用空间。import torch torch.cuda.empty_cache() torch.backends.cuda.max_split_size_mb = 128
内容的提问来源于stack exchange,提问作者FMB
相关产品推荐
相关产品推荐

