快速部署stable diffusion@Ubuntu
在Ubuntu上快速部署stable diffusion模型,您可以遵循以下步骤:
- 安装Anaconda或Miniconda:
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
- 创建一个新的conda环境并激活它:
conda create -n sd python=3.8
conda activate sd
- 安装PyTorch和torchvision(选择与您的系统相匹配的版本):
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
- 克隆stable diffusion仓库:
git clone https://github.com/CompVis/stable-diffusion.git
cd stable-diffusion
- 安装所需的Python依赖项:
pip install -r requirements.txt
- 根据需要配置运行配置文件,例如
denoising_diffusion_unet_finetune_v1.yaml
。 - 运行stable diffusion模型,例如:
python run_sd.py --config denoising_diffusion_unet_finetune_v1.yaml
请注意,上述步骤假设您有一个支持CUDA的NVIDIA GPU,并且安装了相应版本的CUDA。如果您没有GPU或者想在CPU模式下运行,可能需要修改步骤以适应CPU。
评论已关闭