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)启用模型优化,会牺牲很多速度)
然后继续执行脚本