Stable diffusion报Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variab

Stable diffusion 报 Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check 错误 及 排查解决

背景

先说一下我的显卡配置 3060Ti ,当刚开始报这个错误的时候,发现基本上很多博主说的都是在 launch.py 或者 webui-user.bat加一个参数 --precision full --no-half --skip-torch-cuda-test ,webui-user.bat是在 set COMMANDLINE_ARGS=后面加或者launch.py是在index_url = os.environ.get('INDEX_URL', "")加。
加上之后确实不报错了,也能启动起来,但是简单跑了一张图之后,发现cpu奇高,gpu却变化不大,而且速度特别慢,这用cpu跑图也太浪费我的3060Ti了

加参数治标不治本

发现问题

偶然间看到这篇文章,看报错信息和torch有关, 是不是我没用到GPU的torch,用的是cpu的

在这里插入图片描述
原文地址:https://blog.csdn.net/zcs2632008/article/details/127025294

安装步骤

因为之前环境安装的差不多了,能启动,cuda也安装了。所以重点是三四五步

一、NVIDIA一般电脑上都有

二、安装对应版本的CUDA,我是3060Ti对应的12.0版本,默认路径是在C盘下的

三、下载对应版本的cuDNN(它是GPU的加速库)压缩包,解压后会看到几个文件夹 ,内容放到CUDA对应的文件夹(网上很多教程)

四、安装GPU 版的torch 和torchversion 。主要版本不好找,我CUDA是12.0,但是现在最高的才11.8 ,所以用的11.8

可以参考别人的这篇文章:https://blog.csdn.net/qq_43575504/article/details/129559776

五、用命令卸载掉torch 和torchversion

因为最开始大家安装的方式不同,我是最普通的安装方式,有人是直接用的别人封装好的,我说一下我的接下来是怎么操作

  1. 把项目导入PyCharm
  2. 把torch和torchversion两个whl文件放到 \stable-diffusion-webui\venv\Lib\site-packages下面了
  3. 在控制台运行卸载命令和安装命令
    在这里插入图片描述

先卸载再安装,路径改成自己的,别用我的路径

pip uninstall torch #卸载之前安装的torch
pip uninstall torchversion

pip install D:\ruanjian\Stable-Diffusion\stable-diffusion-webui\venv\Lib\site-packages\torch-2.0.1+cu1
18-cp310-cp310-win_amd64.whl

 pip install D:\ruanjian\Stable-Diffusion\stable-diffusion-webui\venv\Lib\site-packages\torchvision-0.1
5.2+cu118-cp310-cp310-win_amd64.whl

六、再启动

把之前添加的--precision full --no-half --skip-torch-cuda-test删掉
再启动
在这里插入图片描述
这时候就没有报错信息了,再跑图,也发现明显快了不少,而且cpu是正常的。
不熟悉可能会耽误几个小时,诸位多点耐心

深夜解决了这个问题,觉得有网友会有跟我一样的问题,所以就写了这篇文章,写的有些简略(每一步都能百度到),望海涵,不过知道问题所在了,就好解决了
  • 32
    点赞
  • 53
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
运行时错误:Torch无法使用GPU;添加--skip-torch-cuda-testCOMMANDLINE_ARGS变量以禁用此检查。对于解决这个问题,你可以尝试以下步骤: 1. 首先,请确保你的系统具备使用GPU的条件。检查你的计算机是否已安装并正确配置了CUDA驱动程序以支持GPU计算。 2. 如果你确定你的系统满足使用GPU的条件,那么你可以按照的建议在命令行中运行相应的命令。运行命令:torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.12.1 cu102 torchvision==0.13.1 cu102 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu102") 。 3. 如果上述步骤无法解决问题,你可以尝试注释或删除错误提示中的那句话,并根据你自己的版本设置相应的语句。可以参考中提供的代码示例。 4. 如果以上步骤仍然无效,你可以尝试在运行webui-user.bat时添加--skip-torch-cuda-test参数,以禁用TorchGPU的检查。这可以通过在命令行中输入命令:webui-user.bat --skip-torch-cuda-test来实现。 请注意,上述步骤是根据引用资料中提供的信息给出的,可能因实际情况而有所不同。如果问题仍然存在,建议你查阅Torch官方文档或在相关论坛上寻求帮助。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [解决在Windows安装stable diffusion遇到“Torch is not able to use GPU”的问题](https://blog.csdn.net/hcaohr/article/details/130122398)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [部署Stable diffusion遇到的一个问题解决](https://blog.csdn.net/MARNieR/article/details/129830832)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值