提高stable diffusion繪圖速度,自主更新Pytorch和xformers方法

1. 前提
找到CUDA與Pytorch對應(yīng)版本(以下使用的是CUDA11.8+Pytorch2.1)
https://developer.nvidia.com/cuda-toolkit-archive

https://pytorch.org/get-started/locally/

安裝Visual Studio 2022,并在其中安裝桌面C++環(huán)境
https://visualstudio.microsoft.com/zh-hans/downloads/選擇第一個


up是自己部署的stable-diffusion,若是其他整合可對應(yīng)看看虛擬環(huán)境是否是venv文件夾
2. 以下你可能會遇到的問題
pytorch無限重復(fù)下載:更換3.10.9的python重啟電腦可以完美解決(重啟電腦)
無法編譯:請確認(rèn)是否將CUDA版本與Pytorch版本對應(yīng)上
下載過慢:更換代理、換源
3. 步驟
1.換源:pip install -i 源url
中國科學(xué)技術(shù)大學(xué) http://pypi.mirrors.ustc.edu.cn/simple/
華中科技大學(xué) http://pypi.hustunique.com/
2. 在stable-diffusion-webui的根目錄下
1.打開powershell(窗口1)

2.進(jìn)入虛擬環(huán)境

3.更新pip所有的軟件包
4.卸載原來的pytorch
5.安裝torch2.0(可以去pytorch官網(wǎng)看看自己對應(yīng)的CUDA版本的下載命令)
6.powershell(窗口1不要關(guān)閉)
3. 找一個位置新建一個文件夾
在新建文件內(nèi)打開powershell(窗口2)

獲取最新的xformers0.17
2.創(chuàng)建虛擬環(huán)境

3.安裝pytorch2.0
4.編譯pytorch2.0與xformers(1小時左右)
5.安裝編譯好的文件
我已經(jīng)編譯好了(除非你的python版本和torch、xformers與其他的人版本一致,否則無法使用他人編譯的文件)

文件在你新建的文件夾/xformers/dist/
在powershell(窗口1)內(nèi)輸入
4. cudnn文件替換(有點用)
cudnn官網(wǎng)下載文件
將壓縮包中的bin內(nèi)的七個dll拖入到 stable-diffusion-webui\venv\Lib\site-packages\torch\lib 中進(jìn)行替換
從兩位UP 視頻中成功學(xué)習(xí)到的
BV17X4y1Q7jq? ? ??BV1DY4y127x6