极客前沿

AI 入门:装好环境,跑通你的第一个模型

2026-05-11 01:33
DEV Machine Learning
查看原文

从零开始,教你安装 Python、PyTorch,并运行一个简单的 AI 图像生成程序。

准备环境:你需要装什么?

要跑 AI 程序,先装好 Python。Python 是 AI 最常用的编程语言,就像盖房子用的砖头。去 python.org 下载最新版,安装时记得勾选“Add Python to PATH”(把 Python 加到系统路径,这样你才能在命令行里直接调用它)。

安装步骤:一步步来

  1. 打开终端(Windows 按 Win+R 输入 cmd,Mac 打开“终端”)。
  2. 输入 python --version,看到版本号说明安装成功。
  3. 安装包管理器 pip(一般自带,可输入 pip --version 检查)。
  4. 安装 PyTorch:去 pytorch.org 选你的系统,复制命令粘贴执行。PyTorch 是 AI 模型的核心库,相当于引擎。
  5. 安装 transformers:输入 pip install transformers。这个库让你方便调用预训练模型。

验证是否成功:跑第一个 AI 程序

创建一个新文件 test.py,用记事本打开,粘贴以下代码:

Tutorial Image
from transformers import pipeline
generator = pipeline('text-generation', model='gpt2')
result = generator('Once upon a time', max_length=50)
print(result[0]['generated_text'])

保存后在终端运行 python test.py。如果看到一段英文故事,恭喜你——AI 环境装好了!

下一步可以做什么

  • 换模型:把 gpt2 换成 distilgpt2bert-base-uncased,体验不同效果。
  • 试试图像生成:安装 diffusers 库(pip install diffusers),用类似代码生成图片。
  • 常见坑:如果安装慢,换国内镜像源(如清华源);如果报错缺库,用 pip install 库名 补装。

内容来源

DEV Machine Learning

发布时间

2026-05-11 01:33

返回 AI技术