极客前沿

AI入门第一步:安装环境并跑通你的第一个模型

2026-05-17 01:30
DEV Beginners
查看原文

零基础学会安装Python、pip和Hugging Face Transformers,用几行代码运行AI模型。

准备环境

在开始之前,你需要安装Python(一种编程语言,AI领域最常用)和pip(Python的包管理器,用来安装其他工具)。

  • 下载Python:访问 python.org,点击下载按钮,选择适合你系统的版本(Windows/macOS/Linux)。
  • 安装时记得勾选“Add Python to PATH”(将Python添加到系统路径),这样你才能在终端里直接使用Python命令。
  • 安装完成后,打开终端(Windows用户按Win+R输入cmd,macOS用户打开“终端”应用),输入 python --version 并回车,如果显示类似“Python 3.x.x”则成功。

安装AI工具库

我们将使用Hugging Face Transformers(一个流行的AI模型库,可以轻松调用预训练模型)。在终端中运行以下命令安装:

  1. 输入 pip install transformers 并回车。等待安装完成,可能需要几分钟。
  2. 如果安装速度慢,可以尝试国内镜像:pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple

跑通你的第一个模型

现在,我们来写一行代码,让AI自动生成一句话。新建一个文本文件,命名为 test.py,用记事本打开,粘贴以下内容:

Tutorial Image
from transformers import pipeline

generator = pipeline('text-generation', model='gpt2')
result = generator('Once upon a time', max_length=30)
print(result[0]['generated_text'])

保存后,在终端中进入该文件所在目录,输入 python test.py 并回车。你将会看到AI续写的故事片段。

如果报错“No module named 'transformers'”,说明pip安装未成功,请重试安装步骤。

下一步可以做什么

  • 尝试修改max_length的值控制输出长度。
  • pipeline('sentiment-analysis')做情感分析。
  • 去Hugging Face官网探索更多免费模型。

内容来源

DEV Beginners

发布时间

2026-05-17 01:30

返回 AI技术