RuntimeError: Torch is not able to use GPU; 解决方案

大家好,我是水滴~~

本文主要介绍在启动 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错误,下面是详细错误:

RuntimeError: Torch is not able to use GPU; 解决方案_第1张图片

解决方案

该错误的提示内容是:Torch 无法使用 GPU,表示你的显示未通过 CUDA 的检测。这时按照提示来添加环境变量,可以将该检测关掉。

编辑 webui-user.bat 文件,在 set COMMANDLINE_ARGS= 的后面添加 --skip-torch-cuda-test 即可,如下图:

RuntimeError: Torch is not able to use GPU; 解决方案_第2张图片

该错误通常发生在 4G 以下显卡,或集显的环境上,此类环境不会使用显卡来工作,而是使用CPU来工作,通常还要增加其他的环境变量,下面是详细参数:

COMMANDLINE_ARGS=--skip-torch-cuda-test --precision full --no-half

但如果你是N卡,显存也是4G以上,那你需要更新一下驱动了。

https://www.nvidia.com/Download/index.aspx

打开上面网址,会自动获取你的显卡型号,然后将下载类型改为:Studio Drive(SD),搜索下载。

RuntimeError: Torch is not able to use GPU; 解决方案_第3张图片

安装好驱动后重新启动 stable-diffusion-webui 即可。

你可能感兴趣的:(AI绘画从入门到精通,AI作画,stable,diffusion,python,Torch,GPU)