IP 属地浙江
琦琦一个爱跳舞 的程序猿
他的动态
torch.cuda.OutOfMemoryError错误表明您在运行模型时遇到了GPU内存不足的问题。这个问题通常是因为模型需要的内存超出了GPU的可用内存。我是在跑stable-diffusion-xl-base-1.0这个模型遇到的这个问题。这个模型包含66亿参数。我用了以下几个策略来解决或缓解这个问题: 降低批量大小(Batch Size): 如果您在训练模型,尝试减少批量大小。这将减少每个训练迭代所需的内存。我是先把batch_size设置为1,观察显存用了多少。优化模…