升级 pytorch2.0 预计可以提升 20% 左右的速度,仅需要升级即可。

安装步骤

1. 进入虚拟环境

为了避免和其他环境库的版本产生冲突,webui 有一套自己的虚拟环境,首先我们需要激活 webui 的虚拟环境,然后在这套虚拟环境中更新。stable-diffusion-webui 文件【Shift 右键】:

🚟 webui 升级 pytorch 2.0 提速 20% - 图1

点击”在此处打开 Powershell 窗口(S)”,进入 PowerShell 终端,执行命令:

  1. $ .\venv\Scripts\Activate.ps1

如图,进入虚拟环境:

🚟 webui 升级 pytorch 2.0 提速 20% - 图2

2. 更新所有软件包

首先安装 pip-review 包,这个包可以帮你更新当前所有的包到最新稳定版本:

  1. $ pip3 install pip-review
  2. # 这行命令交互式更新所有包
  3. $ pip-review --local --interactive

3. 安装 pytorch 2.0 + cuda 11.8

目前官网适配的是 pytorch 2.0 + cuda11.8 的组合。

🚟 webui 升级 pytorch 2.0 提速 20% - 图3

安装命令是:

  1. pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

完成后 pytorch 安装完成,需要安装大概 2.8 个G的文件。


参考

https://www.bilibili.com/video/BV1DY4y127x6/?spm_id_from=333.337.search-card.all.click&vd_source=5b41fc0e1309fb93eff4f49f7ccbd049