windows环境下stable diffusion运行问题汇总
当前 stable diffusion 扩散模型有两个流行的web,一个是 stable-diffusion-webui
,一个是 comfyui
,在windows
环境下12G显存以上的N卡都可以运行。
下面汇总一下运行过程中可能会遇到的问题。
问题1:
Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check
可能是 onnxruntime-gpu
, cuda
, cudnn
之间的版本不匹配
可以通过如下代码验证:
>>> import onnxruntime
>>> onnxruntime.get_device()
'GPU' #表示GPU可用
>>> onnxruntime.get_available_providers()
['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']
如果GPU调用没问题,但是还是报上述错误,可能是torch
版本与sd模型要求的版本不一致,应卸载torch
后重新安装sd模型的指定版本。例如 stable diffusion webui
1.8版本中指定的torch
版本为:
pip install torch==2.1.2 torchvision==0.16.2 --extra-index-url https://download.pytorch.org/whl/cu121
……