极客前沿

AI 入门:一步步装好环境,跑通第一个模型

2026-07-31 01:31
DEV Tutorial
查看原文

从零开始,学会安装 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(一个计算框架,负责让模型运行得更快)。

Tutorial Image
  • 在 Jupyter Notebook 里新建一个代码单元格。
  • 输入以下命令并运行(按 Shift+Enter):
    !pip install transformers torch
  • 如果网速慢,可能要多等几分钟。出现“Successfully installed”就说明装好了。

常见坑:如果报错“pip 不是内部命令”,说明 Python 没添加到 PATH,重装并勾选那个选项即可。

跑通第一个模型:文本生成

现在用 GPT-2(一个能生成文字的小型 AI 模型)来写一句话。

  1. 在 Jupyter Notebook 新单元格里粘贴下面代码:
    from transformers import pipeline
    generator = pipeline('text-generation', model='gpt2')
    result = generator('AI 的未来是', max_length=30)
    print(result)
  2. 运行代码,第一次会下载模型(约 500MB),稍等一下。
  3. 成功后会输出类似 [{'generated_text': 'AI 的未来是...'}] 的结果,说明你成功跑通了第一个 AI 模型!

常见坑:如果下载模型失败,可能是网络问题。可以换个时间再试,或者使用国内镜像源(如添加 --proxy 参数)。

下一步可以做什么

  • 换一个模型试试,比如 model='distilgpt2'(更小更快)。
  • 尝试其他任务,如情感分析:pipeline('sentiment-analysis')('I love this!')
  • 学习更多教程,比如用 Hugging Face(一个分享 AI 模型的网站)上的其他模型。

内容来源

DEV Tutorial

发布时间

2026-07-31 01:31

返回 AI技术