AI 入门:从零搭建环境并跑通第一个模型
手把手教你在本地安装 Python、Jupyter Notebook 和 Transformers 库,并运行一个简单的 AI 文本生成示例。
准备环境:安装 Python
Python 是一种简单易学的编程语言,也是 AI 开发最常用的工具。首先,去 python.org 下载最新版 Python(建议 3.10 或更高)。安装时记得勾选“Add Python to PATH”,这样就能在命令行直接使用。安装完成后,打开终端(Windows 用 cmd 或 PowerShell,Mac/Linux 用终端),输入 python --version,如果显示版本号就说明成功了。
安装 Jupyter Notebook 和常用库
Jupyter Notebook 是一个在浏览器里写代码的工具,特别适合 AI 实验。在终端输入以下命令安装:
pip install jupyter– 安装 Jupyterpip install transformers torch– 安装 Hugging Face 的 Transformers 库(用于加载预训练模型)和 PyTorch(深度学习框架)
如果下载慢,可以加国内镜像:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple jupyter transformers torch。安装完成后,输入 jupyter notebook 启动,浏览器会自动打开一个页面。
第一次跑通:文本生成示例
在 Jupyter 里新建一个 Python 笔记本,复制以下代码并运行:
from transformers import pipeline– 导入流水线工具generator = pipeline('text-generation', model='gpt2')– 加载 GPT-2 模型(第一次会下载,约 500MB)result = generator('Hello, AI is', max_length=30)– 让模型补全文本print(result)– 查看输出
如果看到类似 [{'generated_text': 'Hello, AI is a powerful tool...'}] 的内容,恭喜你,第一个 AI 模型跑通了!
常见坑和下一步
- 报错“No module named 'transformers'”:说明没安装成功,重新运行
pip install transformers并检查网络。 - 模型下载慢:可以设置环境变量
export HF_ENDPOINT=https://hf-mirror.com(Mac/Linux)或set HF_ENDPOINT=https://hf-mirror.com(Windows),使用国内镜像。 - 下一步:试试加载其他模型,比如
bert-base-uncased做文本分类,或者gpt2-medium生成更长的内容。也可以学习用 Hugging Face Datasets 加载数据,训练自己的小模型。
内容来源
freeCodeCamp
发布时间
2026-05-30 01:30