安装指南

本教程面向使用 Diffusers & 昇腾开发者,帮助完成昇腾环境下 Diffusers 的安装。

昇腾环境安装

请根据已有昇腾产品型号及CPU架构等按照 快速安装昇腾环境指引 进行昇腾环境安装,或直接获取对应产品的昇腾环境镜像 cosdt/cann

警告

CANN 最低版本为 8.0.rc1,安装 CANN 时,请同时安装 Kernel 算子包。

Diffusers 安装

Python 环境创建

1# 创建名为 diffusers 的 python 3.10 的虚拟环境
2conda create -y -n diffusers python=3.10
3# 激活虚拟环境
4conda activate diffusers

pip 安装

通过以下指令安装 Diffusers 及 torch-npu:

1pip install diffusers torch==2.2.0 torch-npu==2.2.0 torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

安装校验

执行以下代码,若无任何报错,仅打印模型下载过程,即说明安装成功:

1from diffusers import DiffusionPipeline
2import torch
3
4pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
5pipeline.to("npu")