stable diffusion运行时报错Please reinstall the torchvision that matches your PyTorch install的解决办法

时间:23-09-27 14:50:47

作者:

分类:AI软件教程

    stable diffusion运行时报错Please reinstall the torchvision that matches your PyTorch install的解决办法

  在使用Stable Diffusion时,有时候会遇到报错信息”PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install”。这个报错一般是由于PyTorch和torchvision版本不匹配所致。下面是解决这个问题的步骤:

步骤一:启动虚拟环境

  如果你使用的是Anaconda,可以通过以下命令启动虚拟环境:

conda activate sdwebui && source venv/bin/activate

步骤二:安装旧版本PyTorch

  接下来,你需要安装一个与当前PyTorch版本匹配的旧版本。可以使用以下命令卸载当前的PyTorch,并安装旧版本PyTorch:

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

  请确保你选择的旧版本与你的PyTorch版本匹配。这样可以避免版本不匹配的问题。

步骤三:退出虚拟环境

  完成安装后,你可以通过以下命令退出虚拟环境:

deactivate && conda deactivate

尝试启动WebUI

  现在,你可以尝试重新启动Stable Diffusion的WebUI。如果一切顺利,你应该不再遇到”PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install”的报错信息了。

  希望这些步骤能够帮助你解决问题。