AI 入门:一步步装好环境,跑通第一个模型
从零开始,学会安装 Python、Jupyter Notebook 和 Transformers,并成功运行一个 AI 文本生成示例。
准备环境:装好 Python 和编辑器
AI 开发的第一步是安装 Python(一种编程语言,AI 领域最常用)。
- 去 python.org 下载最新版 Python(建议 3.9 或以上)。
- 安装时记得勾选“Add Python to PATH”(把 Python 加入系统路径,否则后面会找不到命令)。
- 安装完成后,在命令行输入
python --version,看到版本号就表示成功。
接着安装 Jupyter Notebook(一个写代码的网页工具,适合新手边写边看结果)。
- 打开命令行(Windows 按 Win+R 输入 cmd,Mac 打开终端)。
- 输入
pip install jupyter并回车,等待安装完成。 - 安装后输入
jupyter notebook,浏览器会自动打开一个页面,这就是你的编程环境。
安装 AI 库:Transformers 和 PyTorch
要让电脑跑 AI 模型,需要装 Transformers(一个库,里面有很多现成的 AI 模型,像乐高积木一样可以直接用)和 PyTorch(一个计算框架,负责让模型运行得更快)。
- 在 Jupyter Notebook 里新建一个代码单元格。
- 输入以下命令并运行(按 Shift+Enter):
!pip install transformers torch - 如果网速慢,可能要多等几分钟。出现“Successfully installed”就说明装好了。
常见坑:如果报错“pip 不是内部命令”,说明 Python 没添加到 PATH,重装并勾选那个选项即可。
跑通第一个模型:文本生成
现在用 GPT-2(一个能生成文字的小型 AI 模型)来写一句话。
- 在 Jupyter Notebook 新单元格里粘贴下面代码:
from transformers import pipeline
generator = pipeline('text-generation', model='gpt2')
result = generator('AI 的未来是', max_length=30)
print(result) - 运行代码,第一次会下载模型(约 500MB),稍等一下。
- 成功后会输出类似
[{'generated_text': 'AI 的未来是...'}]的结果,说明你成功跑通了第一个 AI 模型!
常见坑:如果下载模型失败,可能是网络问题。可以换个时间再试,或者使用国内镜像源(如添加 --proxy 参数)。
下一步可以做什么
- 换一个模型试试,比如
model='distilgpt2'(更小更快)。 - 尝试其他任务,如情感分析:
pipeline('sentiment-analysis')('I love this!')。 - 学习更多教程,比如用 Hugging Face(一个分享 AI 模型的网站)上的其他模型。
内容来源
DEV Tutorial
发布时间
2026-07-31 01:31