You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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更合理分配显存:
    import torch
    torch.cuda.empty_cache()
    torch.backends.cuda.max_split_size_mb = 128
    
    第一行是清空GPU缓存,第二行是调整显存块拆分大小,减少碎片化,挤出更多可用空间。

内容的提问来源于stack exchange,提问作者FMB

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.09 15:28:12