大家好,我是水滴~~
本文主要介绍在启动 stable-diffusion-webui
时出现的 RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check
问题的解决方案,希望能对你有所帮助。
安装完 stable-diffusion-webui
后,有时会提示 RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check
错误,下面是详细错误:
该错误的提示内容是:Torch 无法使用 GPU
,表示你的显示未通过 CUDA 的检测。这时按照提示来添加环境变量,可以将该检测关掉。
编辑 webui-user.bat
文件,在 set COMMANDLINE_ARGS=
的后面添加 --skip-torch-cuda-test
即可,如下图:
该错误通常发生在 4G 以下显卡,或集显的环境上,此类环境不会使用显卡来工作,而是使用CPU来工作,通常还要增加其他的环境变量,下面是详细参数:
COMMANDLINE_ARGS=--skip-torch-cuda-test --precision full --no-half
但如果你是N卡,显存也是4G以上,那你需要更新一下驱动了。
打开上面网址,会自动获取你的显卡型号,然后将下载类型改为:Studio Drive(SD),搜索下载。
安装好驱动后重新启动 stable-diffusion-webui
即可。