极客前沿

AI 入门:5 分钟装好环境跑通第一个模型

2026-05-20 01:32
DEV Machine Learning
查看原文

从零开始,教你安装 Python 和 Hugging Face,跑通第一个文本生成模型。

准备环境:装好 Python

首先,你需要一个叫 Python 的编程工具(可以理解成让计算机执行指令的语言)。去官网 python.org 下载最新版本(3.10 或更高)。安装时记得勾选“Add Python to PATH”,这样后面用命令才方便。

安装 Hugging Face 库

打开终端(Windows 用命令提示符,Mac 用终端),输入以下命令并按回车:

  • pip install transformers —— 安装 Transformers 库(一个帮你轻松调用各种 AI 模型的工具包)。
  • pip install torch —— 安装 PyTorch(底层计算引擎)。如果电脑有 NVIDIA 显卡,去 pytorch.org 选 CUDA 版本;没有显卡就装 CPU 版。

跑通第一个模型

新建一个 Python 文件(比如 test.py),复制下面代码:

Tutorial Image
from transformers import pipeline
generator = pipeline('text-generation', model='gpt2')
result = generator('Hello, I am', max_length=30)
print(result)

然后运行:python test.py。第一次运行会自动下载模型(约 500MB),稍等一会就会看到 AI 生成的文字。如果报错,常见原因是网络慢,可以设置国内镜像(百度搜索“pip 国内镜像”)。

下一步可以做什么

成功了!现在你可以:

  • 换模型:把 gpt2 改成 distilgpt2(更小更快)或 bert-base-uncased(做填空)。
  • 试试其他任务:pipeline('sentiment-analysis') 做情感分析。
  • 注意:AI 输出可能不准确,永远要用自己的判断核对结果。

内容来源

DEV Machine Learning

发布时间

2026-05-20 01:32

返回 AI技术