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错误，下面是详细错误：

在这里插入图片描述

解决方案

该错误的提示内容是：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以上，那你需要更新一下驱动了。

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

打开上面网址，会自动获取你的显卡型号，然后将下载类型改为：Studio Drive（SD），搜索下载。

在这里插入图片描述

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