Stable Diffusion-1.環境搭建

以下是每一步驟的操作方法:

確認電腦配置

在開始配置之前,確保你的電腦滿足以下最低配置要求:

  • 顯卡:NVIDIA 獨立顯卡,最低為 10 系列,建議使用 40 系列,顯存最低為 4GB,6GB 或更高更佳。
  • 內存:最低為 8GB,建議使用 16GB 或更高。
  • 硬盤:建議有至少 500GB 的可用空間,最佳選擇是固態硬盤。
  • 操作系統:支持 Windows 10/11、macOS(僅限 Apple Silicon)和 Linux 系統。

如果你沒有滿足以上配置的電腦,可以考慮購買云主機,如騰訊 GPU 云服務器。如果無法使用獨立顯卡和云服務,你也可以修改啟動配置,使用 CPU 渲染(兼容性強,但出圖速度較慢,需要至少 16GB 內存)。

環境準備

  1. 確保你的電腦上有NVIDIA顯卡,并且已經安裝了顯卡驅動。你可以通過查看設備管理器或者執行命令nvidia-smi來確認是否有獨立顯卡可用 [1].
  2. 下載并安裝CUDA Toolkit,選擇與你的顯卡驅動版本對應的CUDA版本 [1].
  3. 下載并安裝cuDNN,選擇與你安裝的CUDA版本對應的cuDNN版本 [1].
  4. 下載并安裝Anaconda,選擇與你的操作系統和Python版本對應的Anaconda版本 [1].

安裝pytorch虛擬環境

  1. 打開命令行終端,執行以下命令創建一個Python 3.10.6版本的虛擬環境:
conda create --prefix <虛擬環境路徑> python==3.10.6

其中,<虛擬環境路徑>是你想要創建的虛擬環境的路徑。
2. 激活虛擬環境:

conda activate <虛擬環境路徑>

安裝pytorch

  1. 在激活的虛擬環境中,執行以下命令安裝pytorch:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

  1. 驗證pytorch是否安裝成功:
python
import torch
x = torch.rand(5, 3)
print(x)
torch.cuda.is_available()

如果沒有報錯,并且能夠輸出隨機張量和CUDA的可用性,則表示pytorch安裝成功。

安裝Stable Diffusion Web UI

  1. 使用Git克隆Stable Diffusion Web UI倉庫:
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

  1. 進入項目目錄:
cd stable-diffusion-webui

  1. 在激活的虛擬環境中,執行以下命令安裝依賴包:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
conda install --yes --file requirements.txt

  1. 啟動Stable Diffusion Web UI:
python launch.py

首次啟動會下載一些模型和權重文件,請耐心等待。


Learn more:

  1. 【AI繪畫】本地部署Stable Diffusion Web UI_AI作畫_測試開發小記-蘇州城市開發者社區
  2. stable-diffusion-webui手動安裝詳細步驟(以及報錯解決、踩坑)-CSDN博客
  3. pytorch:不同版本的CUDA和CUDNN安裝教程-騰訊云開發者社區-騰訊云
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容