解决stable diffusion遇到“RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to .”的问题

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; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check

出现这个问题就对了,不出现就不用看本篇文章了。

SD默认是仅支持N卡的,也是推荐使用N卡的,这里主要是torch会检查CUDA(这个是N卡先关的东西,我们A卡是没有的)。

网上很多教程都说的是在执行脚本里加 --skip-torch-cuda-test参数,确实加了就不报错了,但推荐这种做法的人都是敷衍解决问题。加了这个后不会再使用GPU算力,而是使用处理器的CPU算力!!! 不信,你可以试试。一试一个不吱声

正确姿势是加入参数 --use-directml

(ps:如果显存过低,可再参考参数

--medvram: 为低显存(比如:4G)启用模型优化,会牺牲一点速度

--lowvram: 为极低显存(比如:2G)启用模型优化,会牺牲很多速度)

然后继续执行脚本

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值