极客前沿

AI入门:从零安装到跑通第一个模型

2026-06-29 01:31
DEV Machine Learning
查看原文

手把手教你安装Python和Transformers库,并运行你的第一个AI模型。

准备工作:你需要什么

在开始之前,确保你的电脑能联网,并且有至少4GB内存。我们将使用Python(一种编程语言,就像电脑的通用语言)来运行AI模型。如果你没有Python,别担心,下面会教你安装。

第一步:安装Python

  1. 打开浏览器,访问 python.org,点击“Downloads”下载最新版Python(建议3.9或更高)。
  2. 运行安装包,务必勾选“Add Python to PATH”(这样电脑才能找到Python),然后点击“Install Now”。
  3. 安装完成后,打开命令提示符(Windows按Win+R,输入cmd回车;Mac打开终端),输入 python --version,如果显示版本号,说明安装成功。

第二步:安装Transformers库

Transformers是一个工具包,让AI模型更容易使用。我们通过pip(Python的安装助手)来安装。

Tutorial Image
  1. 在命令提示符中,输入 pip install transformers,按回车。等待下载完成,中间可能会提示“Successfully installed”。
  2. 如果遇到“pip不是内部命令”,说明Python没加到PATH,重新安装Python并勾选那个选项。

第三步:运行你的第一个AI模型

我们将使用一个文本生成模型(能根据你给的开头自动写句子)。

  1. 创建一个新文件,命名为 test.py,用记事本打开。
  2. 复制以下代码:
    from transformers import pipeline
    generator = pipeline('text-generation', model='gpt2')
    result = generator('Hello, I am learning AI', max_length=30)
    print(result)
  3. 保存文件,然后在命令提示符中,进入文件所在文件夹(例如 cd Desktop),输入 python test.py 运行。
  4. 第一次运行会下载模型(约500MB),耐心等待。成功后,你会看到类似 [{'generated_text': 'Hello, I am learning AI and ...'}] 的输出。

常见坑和下一步

  • 下载慢:可以设置镜像源,例如 pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 内存不足:换小模型,如 distilgpt2
  • 下一步:尝试修改输入文本,或试试其他任务(如情感分析),访问 huggingface.co/models 探索更多模型。

内容来源

DEV Machine Learning

发布时间

2026-06-29 01:31

返回 AI技术