Python 官方文档:入门教程 => 点击学习
Stable diffusion 报 Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS vari
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也安装了。所以重点是三四五步
可以参考别人的这篇文章:Https://blog.csdn.net/qq_43575504/article/details/129559776
因为最开始大家安装的方式不同,我是最普通的安装方式,有人是直接用的别人封装好的,我说一下我的接下来是怎么操作
\stable-diffusion-webui\venv\Lib\site-packages
下面了先卸载再安装,路径改成自己的,别用我的路径
pip uninstall torch #卸载之前安装的torchpip uninstall torchversionpip install D:\ruanjian\Stable-Diffusion\stable-diffusion-webui\venv\Lib\site-packages\torch-2.0.1+cu118-cp310-cp310-win_amd64.whl pip install D:\ruanjian\Stable-Diffusion\stable-diffusion-webui\venv\Lib\site-packages\torchvision-0.15.2+cu118-cp310-cp310-win_amd64.whl
把之前添加的--precision full --no-half --skip-torch-cuda-test
删掉
再启动
这时候就没有报错信息了,再跑图,也发现明显快了不少,而且cpu是正常的。
不熟悉可能会耽误几个小时,诸位多点耐心
来源地址:https://blog.csdn.net/qq_44790702/article/details/130997831
--结束END--
本文标题: Stable diffusion报Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variab
本文链接: https://www.lsjlt.com/news/390835.html(转载时请注明来源链接)
有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
2024-03-01
2024-03-01
2024-03-01
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
回答
回答
回答
回答
回答
回答
回答
回答
回答
回答
0