Stable Diffusion升級(jí)torch2.1.0.dev+cu121 cudnn8.9.2.26 xformers0.0

原來的
version:v1.3.2? python:3.10.6? torch:2.0.1+cu118? cudnn:8.8.1.3? xformers:0.0.17?? gradio:3.32.0? checkpoint:cc6cb27103
之前為了提速?gòu)腸uda12.1.1+cudnn8.9.2.26裝回cuda11.8+cudnn8.8.1.3
現(xiàn)在支持cuda12.1.1的torch2.1.0有了,就差xformers,自己來編譯一下。

1.卸載原來的cuda11.8(刪除默認(rèn)路徑C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8殘留的文件夾,里面包含cudnn8.8的文件)
?
2.下載并安裝cuda12.1.1
https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_531.14_windows.exe
?
3.下載cudnn8.9.2.26
https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.2/local_installers/12.x/cudnn-windows-x86_64-8.9.2.26_cuda12-archive.zip

4.下載并安裝Microsoft C++生成工具(默認(rèn)安裝即可)
https://visualstudio.microsoft.com/zh-hans/visual-cpp-build-tools/


5.打開miniconda3
6.輸入conda activate sd(激活名為sd的python虛擬環(huán)境)

7.輸入cd d:\sdweb\(切換到Stable diffusion目錄)
8.再輸入pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/(切換清華鏡像源)

9.輸入.\venv\Scripts\Activate.ps1(進(jìn)入sd虛擬環(huán)境)


###(此處跟進(jìn)度無(wú)關(guān))從這里開始,python的虛擬環(huán)境我由sd換成sdwebui,Stable diffusion的目錄由D:\sdweb換成D:\sd ,我電腦裝太多個(gè)版本,避免看亂了,在這里說一下,也打碼一下###

10.?pip uninstall xformers(按y卸載舊版xformers0.17)

11.?pip uninstall torch(按y卸載舊版torch 2.0.1+cu118)

12.?pip install ninja(安裝ninja編譯工具)

13.自動(dòng)更新所有python包(依次輸入)
pip3 install pip review (安裝pip review)
pip review (查看可更新的包)
pip review -- auto (自動(dòng)更新所有包)

14.獲取命令并安裝?對(duì)應(yīng)cuda12.1平臺(tái)的PyTorch2.1
pip3 install --pre torch torchvision torchaudio --index-url?https://download.pytorch.org/whl/nightly/cu121

這里要點(diǎn)時(shí)間下載

15.克隆xformers0.0.21到Stable diffusion根目錄(我的目錄已改成D:\sd 前面是D:\sdweb)
git clone https://ghproxy.com/https://github.com/facebookresearch/xformers.git

16.?進(jìn)入克隆好的xformers里的.git目錄(此目錄隱藏狀態(tài)),按圖操作,字不打了-。-


17.切換到克隆好的xformers目錄
cd xformers

18.在submodule庫(kù)里升級(jí)一下,怕失敗或者真的失敗了,多執(zhí)行幾次確認(rèn)一下
git submodule update --init --recursive

19.?導(dǎo)包
pip install -r requirements.txt
20.安裝python wheel文件
pip install wheel

21.編譯PyTorch2.1+cuda12.1+xformers0.0.21(我花了5分鐘不到,因設(shè)備而異。)
python setup.py build

22.項(xiàng)目打包(10秒就好了)
python setup.py bdist_wheel

23.?找到并安裝編譯好的.whl文件
pip install 文件路徑...

24.?完成 退出

25.?頂部第2.下載的cudnn解壓并把bin文件夾里面的7個(gè)文件替換
D:\sd\venv\Lib\site-packages\torch\lib\

我兩個(gè)版本都編譯了,沒有用同個(gè)編譯文件。迭代次數(shù)還是比較穩(wěn)定的。



