相關博文

1. stable-diffusion-webui的基礎功能手動安裝，了解代碼結構、依賴、模型出處

安裝成功結果

主要參考

• cuda11.7下載-https://developer.nvidia.com/cuda-toolkit-archive
• cudnn8.6 for 11.x : https://developer.nvidia.com/rdp/cudnn-archive
• dreambooth\lora 訓練環(huán)境：
  • 最原始的命令端 https://github.com/kohya-ss/sd-scripts
  • GUI端口 https://github.com/bmaltais/kohya_ss
• 環(huán)境安裝參考：win11 系統(tǒng) GPU版本pytorch、CUDA、anaconda 、pycharm詳細安裝教程

一、顯卡驅動相關

驅動版本自帶cuda 12.0 與pytorch不兼容

驅動要安裝官網(wǎng)4090自帶的：528.02 cuda12.0 (運維安裝跳過)

pytorch 需要cuda12.0以下

自定義安裝

取消安裝驅動

明確安裝位置，好添加到系統(tǒng)路徑

安裝成功和失敗的

cudnn 下載與安裝

• https://developer.nvidia.com/rdp/cudnn-archive

改名放到cuda安裝目錄下，并添加到環(huán)境變量

根目錄

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\

結果

最終環(huán)境變量

肯定有些冗余

二、安裝 python3.10 + pytorch

下載安裝miniconda （python 3.10.9）

官網(wǎng)直接選擇最新版本 ： https://docs.conda.io/en/latest/miniconda.html

添加到系統(tǒng)環(huán)境

計算機—》屬性—》高級系統(tǒng)設置----》環(huán)境變量—》系統(tǒng)變量—》path

根據(jù)自己位置改

D:\miniconda310\Scripts
D:\miniconda310

安裝pytorch

• https://pytorch.org/
  

配置pip的國內(nèi)源并升級

pip config set global.index-url  https://mirror.baidu.com/pypi/simple
pip config set global.trusted-host mirror.baidu.com
python -m pip install --upgrade pip

查找對應的pytorch

lora dreambooth推薦的環(huán)境：https://github.com/bmaltais/kohya_ss/blob/master/setup.bat

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

需要15分鐘左右時間

三、其他python依賴包

安裝requirements

–upgrade使用最新版本的pip
進入 kohya_ss-master的位置

pip install --upgrade -r requirements.txt

ERROR: Cannot find command ‘git’ - do you have ‘git’ installed and in your PATH?

https://gitforwindows.org/

conda install git

git不是內(nèi)部外部命令，添加到系統(tǒng)環(huán)境

• https://www.cnblogs.com/ldq678/p/13287924.html
  找到自己的路徑
  

安裝過程截圖

開始安裝

成功

安裝 xformer

• https://github.com/facebookresearch/xformers

conda install xformers -c xformers
or
pip install -U xformers

本項目特定環(huán)境安裝（不然訓練報錯，必須裝）

copy /y .\bitsandbytes_windows\*.dll D:\miniconda310\Lib\site-packages\bitsandbytes\
copy /y .\bitsandbytes_windows\cextension.py D:\miniconda310\Lib\site-packages\bitsandbytes\cextension.py
copy /y .\bitsandbytes_windows\main.py D:\miniconda310\Lib\site-packages\bitsandbytes\cuda_setup\main.py

accelerate config  

配置

erro 訓練時報錯 No model named ‘triton’

鏈接：https://pan.baidu.com/s/1shX-cU-R-0_gKhiTM0VXSw
提取碼：6666

pip install triton-2.0.0-cp310-cp310-win_amd64.whl

四、啟動訓練界面

python D:\kohya_ss-master\kohya_gui.py

附錄

本項目cudnn8.6 特殊安裝

4090使用cudnn8.6能加速50%

下載后解壓，放入文件夾

將cudnn保存到虛擬環(huán)境中

python D:\kohya_ss-master\tools\cudann_1.8_install.py

文章來源地址http://www.zghlxwxcb.cn/news/detail-464021.html

到了這里，關于【stable-diffusion】4090顯卡下dreambooth、lora、sd模型微調(diào)的GUI環(huán)境安裝（cuda驅動、pytorch、xformer）的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網(wǎng)！