返回

AI 作画革命：掌握 Linux 上 Stable Diffusion 2.0 的安装指南

后端

前言

随着 Stable Diffusion 2.0 的到来，AI 作画迎来了一个激动人心的新篇章。这个强大的文本到图像模型释放了无穷的创造潜力，让您可以将您的想象力转化为令人惊叹的视觉杰作。然而，在 Linux 上安装 Stable Diffusion 2.0 可能是一项艰巨的任务，需要具备一定的技术知识。

为了帮助您克服这一挑战，我们编制了这份全面的教程，提供逐步指导，让您轻松安装和配置 Stable Diffusion 2.0。我们将深入探讨 Linux 特有的细微差别，并提供明确的说明和有用的示例，让您的安装过程顺利无阻。

先决条件

在继续进行之前，确保您的系统满足以下先决条件：

  • Ubuntu 22.04 或更高版本
  • NVIDIA GPU 具有至少 8GB 显存
  • CUDA 11.6 或更高版本
  • PyTorch 1.13 或更高版本
  • Hugging Face Transformers 4.29 或更高版本

安装过程

1. 安装依赖项

首先，您需要安装 Stable Diffusion 2.0 所需的依赖项。运行以下命令：

sudo apt update && sudo apt install wget unzip git python3-pip

2. 安装 CUDA 和 cuDNN

接下来，安装 CUDA 和 cuDNN 以支持 NVIDIA GPU 加速：

CUDA 11.6：

wget https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda-repo-ubuntu-1804-11-6-local_11.6.1-470.42.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu-1804-11-6-local_11.6.1-470.42.06-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu-1804-11-6-local/7fa2af80.pub
sudo apt update
sudo apt install cuda

cuDNN 8.6.1：

wget https://developer.download.nvidia.com/compute/cudnn/secure/v8.6.1/cudnn-11.6-linux-x64-v8.6.1.32.tgz
tar -xzf cudnn-11.6-linux-x64-v8.6.1.32.tgz
sudo cp cuda/include/* /usr/local/include/
sudo cp cuda/lib64/* /usr/local/lib64/

3. 安装 PyTorch 和 Hugging Face Transformers

安装 PyTorch 和 Hugging Face Transformers：

pip install torch torchvision torchaudio torchtext
pip install transformers

4. 克隆 Stable Diffusion 2.0 存储库

克隆 Stable Diffusion 2.0 存储库：

git clone https://github.com/CompVis/stable-diffusion
cd stable-diffusion

5. 安装 Stable Diffusion 2.0

安装 Stable Diffusion 2.0：

pip install -e .

配置和使用

现在，Stable Diffusion 2.0 已成功安装在您的系统上。您可以通过以下步骤配置和使用它：

1. 下载模型权重：

cd models
wget https://huggingface.co/CompVis/stable-diffusion-v2-1-base/resolve/main/sd-v2-1-base.ckpt

2. 运行示例脚本：

在 scripts 目录中，运行以下命令生成图像：

python scripts/txt2img.py --prompt "A photo of a cat"

结论

恭喜！您已成功在 Linux 系统上安装和配置了 Stable Diffusion 2.0。现在，您拥有了将想象力转化为视觉艺术的力量。探索 Stable Diffusion 2.0 的无限可能性，创建令人惊叹的图像，并踏上激动人心的 AI 辅助图像生成之旅。

如果您在安装过程中遇到任何问题，请随时提出问题，我们会尽力提供帮助。祝您 AI 作画之旅愉快！