AI绘画-Stable Diffusion解决报错教程-显存不足​

哈喽！伙伴们，我是小智，你们的AI向导。欢迎来到每日的AI学习时间。今天，我们将一起深入AI的奇妙世界，探索“AI绘画-Stable Diffusion解决报错教程-显存不足​”，并学会本篇文章中所讲的全部知识点。还是那句话“不必远征未知，只需唤醒你的潜能！”跟着小智的步伐，我们终将学有所成，学以致用，并发现自身的更多可能性。话不多说，现在就让我们开始这场激发潜能的AI学习之旅吧。

AI绘画-Stable Diffusion解决报错教程-显存不足​：

显存不足​

(base) D:openai.wikistable-diffusion-webui>webui-user.bat​
venv “D:openai.wikistable-diffusion-webuivenvScriptsPython.exe”​
Python 3.10.9 | packaged by Anaconda, Inc. | (main, Mar 8 2023, 10:42:25) [MSC v.1916 64 bit (AMD64)]​
Commit hash: a9fed7c364061ae6efb37f797b6b522cb3cf7aa2​
Installing requirements for Web UI​
Launching Web UI with arguments:​
No module ‘xformers’. Proceeding without it.​
Loading weights [ad2a33c361] from D:openai.wikistable-diffusion-webuimodelsStable-diffusionv2-1_768-ema-pruned.ckptCreating model from config: D:openai.wikistable-diffusion-webuirepositoriesstable-diffusion-stability-aiconfigsstable-diffusionv2-inference-v.yaml​
LatentDiffusion: Running in v-prediction mode​
DiffusionWrapper has 865.91 M params.​
loading stable diffusion model: OutOfMemoryError​
Traceback (most recent call last):​
File “D:openai.wikistable-diffusion-webuiwebui.py”, line 136, in initialize​
modules.sd_models.load_model()​
File “D:openai.wikistable-diffusion-webuimodulessd_models.py”, line 441, in load_model​
sd_model.to(shared.device)​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagespytorch_lightningcoremixinsdevice_dtype_mixin.py”, line 113, in to​
return super().to(*args, **kwargs)​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagestorchnnmodulesmodule.py”, line 989, in to​
return self._apply(convert)​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagestorchnnmodulesmodule.py”, line 641, in _apply​
module._apply(fn)​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagestorchnnmodulesmodule.py”, line 641, in _apply​
module._apply(fn)​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagestorchnnmodulesmodule.py”, line 641, in _apply​
module._apply(fn)​
[Previous line repeated 4 more times]​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagestorchnnmodulesmodule.py”, line 664, in _apply​
param_applied = fn(param)​
File “D:openai.wikistable-diffusion-webuivenvlibsite-packagestorchnnmodulesmodule.py”, line 987, in convert​
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)​
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.66 GiB already allocated; 0 bytes free; 1.71 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​
Stable diffusion model failed to load, exiting​
请按任意键继续。 . .​

根据错误信息显示，Stable Diffusion 模型加载时遇到了 CUDA 显存不足的错误，简单点来说就是显卡不行，没有足够的显存空间分配给该模型。​
解决方法 1:​
换个好一些的显卡，至少英伟达 2060 以上。​

解决方法 2:​
开启低显存模式，开启低显存模式的方法如下。​
在 stable-diffusion-webui 文件夹下找到 webui-user.bat，用文本或代码编辑器打开该文件夹，可以看到如下内容。​
@echo off​
set PYTHON=​
set GIT=​
set VENV_DIR=​
set COMMANDLINE_ARGS=​
call webui.bat​
我们只需要修改set COMMANDLINE_ARGS=部分即可，该部分内容是启动参数。​

低于 3G 显存​
如果你的显卡显存不足 3G，可以在set COMMANDLINE_ARGS=的后面添加参数–lowvram –always-batch-cond-uncond之后保存即可，变更之后的文件内容如下。​
@echo off
set PYTHON=​
set GIT=​
set VENV_DIR=​
set COMMANDLINE_ARGS=–lowvram –always-batch-cond-uncond​
call webui.bat​

4G 显存​
如果你的显卡只有 4G 显存，可以在set COMMANDLINE_ARGS=的后面添加参数–precision full –no-half –lowvram –always-batch-cond-uncond之后保存即可，变更之后的文件内容如下。​
@echo off​
set PYTHON=​
set GIT=​
set VENV_DIR=​
set COMMANDLINE_ARGS=–precision full –no-half –lowvram –always-batch-cond-uncond​
call webui.bat​

低于 5G 显存​
如果你的显卡显存不足 5G，可以在set COMMANDLINE_ARGS=的后面添加参数–medvram之后保存即可，变更之后的文件内容如下。​
@echo off​
set PYTHON=​
set GIT=​
set VENV_DIR=​
set COMMANDLINE_ARGS=–medvram​
call webui.bat​

6G 显存​
如果你的显卡只有 6G 显存，可以在set COMMANDLINE_ARGS=的后面添加参数–precision full –no-half –medvram之后保存即可，变更之后的文件内容如下。​
@echo off​
set PYTHON=​
set GIT=​
set VENV_DIR=​
set COMMANDLINE_ARGS=–precision full –no-half –medvram​
call webui.bat​
​

嘿，伙伴们，今天我们的AI探索之旅已经圆满结束。关于“AI绘画-Stable Diffusion解决报错教程-显存不足​”的内容已经分享给大家了。感谢你们的陪伴，希望这次旅程让你对AI能够更了解、更喜欢。谨记，精准提问是解锁AI潜能的钥匙哦！如果有小伙伴想要了解学习更多的AI知识，请关注我们的官网“AI智研社”，保证让你收获满满呦！