运行novelAI cuda.OutOfMemoryError GPU内存不足 但是GPU内存有空间

我参考的教程:
bilibili AI作画保姆级教程来了!逆天,太强了!
AI杀疯了,NovelAI开源视频教程来了,一键安装

但是我出现了torch.cuda.OutOfMemoryError的错误,尤其是我使用nvidia-smi查看发现gpu内存没有被占用。
运行novelAI cuda.OutOfMemoryError GPU内存不足 但是GPU内存有空间_第1张图片

在网上查怎么办,大多解决方法是中止进程释放资源,但是我根本没有进程(no running processes found)
我灵机一动,意识到是不是自己的GPU显存太小,只有2G。确实如此。
Novelai 保姆级免费部署和最全使用教程 (含资源与常见错误大全)(一) 三.常见错误

如果显存小于等于4GB可通过低速模式进行运行,请参考以下建议:
翻译自stable-diffusion-webui wiki)
如果您有 4GB VRAM,并且想要制作 512x512(或最高 640x640)的图像,请使用 。–medvram
如果您有 4GB VRAM 并希望制作 512x512 图像,但出现内存不足错误,请改用 。–medvram–medvram --opt-split-attention
如果您有 4GB VRAM 并想要制作 512x512 图像,但仍然收到内存不足错误,请改用。–lowvram --always-batch-cond-uncond --opt-split-attention
如果您有 4GB 的 VRAM,并且希望使图像比 使用 时更大,请使用 。–medvram–lowvram --opt-split-attention
如果您有更多 VRAM,并且想要制作比通常更大的图像(例如,1024x1024 而不是 512x512),请使用 。您也可以使用,但效果可能几乎不明显。–medvram --opt-split-attention–lowvram
否则,请勿使用其中任何一个。

我在运行的时候输入:
python launch.py --lowvram --always-batch-cond-uncond --opt-split-attention 成功运行。

你可能感兴趣的:(novelAI,深度学习)