以下是每一步驟的操作方法:
確認電腦配置
在開始配置之前,確保你的電腦滿足以下最低配置要求:
- 顯卡:NVIDIA 獨立顯卡,最低為 10 系列,建議使用 40 系列,顯存最低為 4GB,6GB 或更高更佳。
- 內存:最低為 8GB,建議使用 16GB 或更高。
- 硬盤:建議有至少 500GB 的可用空間,最佳選擇是固態硬盤。
- 操作系統:支持 Windows 10/11、macOS(僅限 Apple Silicon)和 Linux 系統。
如果你沒有滿足以上配置的電腦,可以考慮購買云主機,如騰訊 GPU 云服務器。如果無法使用獨立顯卡和云服務,你也可以修改啟動配置,使用 CPU 渲染(兼容性強,但出圖速度較慢,需要至少 16GB 內存)。
環境準備
- 確保你的電腦上有NVIDIA顯卡,并且已經安裝了顯卡驅動。你可以通過查看設備管理器或者執行命令
nvidia-smi
來確認是否有獨立顯卡可用 [1]. - 下載并安裝CUDA Toolkit,選擇與你的顯卡驅動版本對應的CUDA版本 [1].
- 下載并安裝cuDNN,選擇與你安裝的CUDA版本對應的cuDNN版本 [1].
- 下載并安裝Anaconda,選擇與你的操作系統和Python版本對應的Anaconda版本 [1].
安裝pytorch虛擬環境
- 打開命令行終端,執行以下命令創建一個Python 3.10.6版本的虛擬環境:
conda create --prefix <虛擬環境路徑> python==3.10.6
其中,<虛擬環境路徑>
是你想要創建的虛擬環境的路徑。
2. 激活虛擬環境:
conda activate <虛擬環境路徑>
安裝pytorch
- 在激活的虛擬環境中,執行以下命令安裝pytorch:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
- 驗證pytorch是否安裝成功:
python
import torch
x = torch.rand(5, 3)
print(x)
torch.cuda.is_available()
如果沒有報錯,并且能夠輸出隨機張量和CUDA的可用性,則表示pytorch安裝成功。
安裝Stable Diffusion Web UI
- 使用Git克隆Stable Diffusion Web UI倉庫:
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
- 進入項目目錄:
cd stable-diffusion-webui
- 在激活的虛擬環境中,執行以下命令安裝依賴包:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
conda install --yes --file requirements.txt
- 啟動Stable Diffusion Web UI:
python launch.py
首次啟動會下載一些模型和權重文件,請耐心等待。
Learn more: