AI入门第一步:装好环境,跑通你的第一个模型
零基础学会安装Python、Jupyter和Transformers库,并运行一个简单的AI文本生成模型。
准备环境:你需要装什么?
要跑AI模型,首先得准备一个“工作台”。我们只需要三样东西:Python(一种编程语言,像电脑的“普通话”)、Jupyter Notebook(一个写代码的笔记本,能边写边看结果)和Transformers库(一个装满了现成AI模型的工具箱)。
- Python:去官网 python.org 下载最新版(3.10以上),安装时记得勾选“Add Python to PATH”(否则后面会报找不到命令)。
- Jupyter Notebook:打开命令行(Windows按Win+R输入cmd,Mac打开终端),输入
pip install jupyter回车,等它装完。 - Transformers库:同样在命令行输入
pip install transformers并回车。
常见坑:如果提示“pip不是内部命令”,说明Python没加到PATH里,重装时勾选那个选项就好。
安装步骤:一步一步来
- 验证Python安装:在命令行输入
python --version,如果显示版本号(比如Python 3.11.2),说明成功。 - 启动Jupyter:在命令行输入
jupyter notebook,浏览器会自动打开一个页面,像文件夹一样。 - 新建笔记本:点右上角“New” → “Python 3”,就会打开一个空白格子(叫“单元格”)。
- 安装Transformers:在单元格里输入
!pip install transformers,按Shift+Enter运行,看到“Successfully installed”就搞定。
常见坑:Jupyter页面打不开?检查命令行是否还开着,别关掉。如果报错端口被占用,换一个端口:jupyter notebook --port=8889。
验证是否成功:跑通你的第一个模型
我们来用一个小模型试试手:文本生成——给它一句话,它帮你续写。这背后是GPT-2(一个能写文章的AI,就像简化版的ChatGPT)。
- 在单元格里输入以下代码:
from transformers import pipelinegenerator = pipeline('text-generation', model='gpt2')result = generator('AI is the future because', max_length=30)print(result[0]['generated_text']) - 按Shift+Enter运行。第一次运行会下载模型(几百MB),等几分钟。
- 看到输出一段英文(比如“AI is the future because it can help us solve complex problems...”),就说明你成功了!
常见坑:下载慢?可以换成国内镜像:在代码前加 import os; os.environ['HF_ENDPOINT']='https://hf-mirror.com'。
下一步可以做什么
- 换模型试试:把
'gpt2'换成'distilgpt2'(更小更快)或'bert-base-uncased'(做句子填空)。 - 学点Python基础:推荐免费教程《Python官方入门》或《笨办法学Python》。
- 探索更多任务:Transformers库还支持翻译、问答、情感分析,把
'text-generation'换成'sentiment-analysis'就能做情感判断。
现在你已经入门了!记住:遇到问题先看报错信息,复制到搜索引擎里,大部分坑都有人踩过。祝玩得开心!
内容来源
DEV Tutorial
发布时间
2026-06-20 01:31