AI 绘图入门:一步步安装并跑通你的第一个模型
读完本文,你将学会如何安装 Python、配置环境,并运行一个简单的 AI 绘图模型。
准备环境:装好 Python 和 pip
首先,你需要在电脑上安装 Python(一种编程语言,AI 工具大多用它编写)。去 python.org 下载最新版(推荐 3.10 或 3.11),安装时记得勾选“Add Python to PATH”选项。装好后,打开终端(Windows 按 Win+R,输入 cmd;Mac 打开“终端”),输入 python --version,看到版本号就成功了。pip 是 Python 的包管理器,用来安装各种库,通常随 Python 一起安装,输入 pip --version 验证。
安装 AI 绘图库:用 diffusers 和 transformers
我们使用 Hugging Face 的 diffusers(一个让 AI 绘图变得简单的工具库)和 transformers(处理文字和图像的库)。在终端中依次执行以下命令:
pip install diffusers transformers accelerate torch(torch 是 PyTorch,AI 计算的核心引擎)- 如果下载慢,可以加上国内镜像:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple diffusers transformers accelerate torch - 等待安装完成,如果出现红色报错,通常是因为网络或权限问题,可以尝试用管理员身份运行终端(Windows 右键“以管理员身份运行”)。
跑通第一个模型:生成一张图片
新建一个 Python 文件,比如 test.py,复制以下代码:
from diffusers import StableDiffusionPipeline
import torch
model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
prompt = "a cute cat sitting on a chair"
image = pipe(prompt).images[0]
image.save("cat.png")然后在终端运行 python test.py。第一次运行会下载模型(约 2GB),请保持网络畅通。如果电脑没有 NVIDIA 显卡(即没有 CUDA),代码会自动用 CPU 运行,但速度会慢很多。生成成功后,文件夹里会出现 cat.png,打开看看!
常见问题与下一步
- 显存不足:如果报错“CUDA out of memory”,可以减小图片尺寸,在代码中加
pipe.safety_checker = None关闭安全检查也能省点显存。 - 模型下载失败:检查网络,或者手动从 Hugging Face 官网下载模型放到本地路径。
- 下一步:尝试修改
prompt里的文字生成不同图片;或者安装 Gradio(一个快速搭建网页界面的库),用pip install gradio然后写几行代码就能在浏览器里玩 AI 绘图了。
内容来源
DEV Machine Learning
发布时间
2026-07-23 01:37