stable-diffusion-webui踩坑分享

1,924 阅读1分钟

很多人会遇到一个问题 # Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check 如果你真的按网上或者提示里的照做了,加了 --skip-torch-cuda-test这个参数 那你就真的冤大头了,实际上如果你是n卡的话,出现这个问题很大可能是装错了PyTorch版本 验证方法如下

import torch
print(torch.__version__)
print(torch.cuda.is_available())

预期输出

1.13.1+cu117 # Your CUDA version
True

如果第一行版本中出现cpu字样的话,直接去重装一个正确的版本就好了

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117

参考文献 [1]torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-CSDN博客