升级 pytorch2.0 预计可以提升 20% 左右的速度,仅需要升级即可。
安装步骤
1. 进入虚拟环境
为了避免和其他环境库的版本产生冲突,webui 有一套自己的虚拟环境,首先我们需要激活 webui 的虚拟环境,然后在这套虚拟环境中更新。stable-diffusion-webui 文件【Shift 右键】:
点击”在此处打开 Powershell 窗口(S)”,进入 PowerShell 终端,执行命令:
$ .\venv\Scripts\Activate.ps1
如图,进入虚拟环境:
2. 更新所有软件包
首先安装 pip-review 包,这个包可以帮你更新当前所有的包到最新稳定版本:
$ pip3 install pip-review
# 这行命令交互式更新所有包
$ pip-review --local --interactive
3. 安装 pytorch 2.0 + cuda 11.8
目前官网适配的是 pytorch 2.0 + cuda11.8 的组合。
安装命令是:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
完成后 pytorch 安装完成,需要安装大概 2.8 个G的文件。