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

发布时间:2024年01月19日

大家好,我是水滴~~

本文主要介绍在启动 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 即可。

文章来源:https://blog.csdn.net/weixin_67276852/article/details/135692251
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。