极客前沿

AI入门:手把手教你装好环境并跑通第一个模型

2026-06-03 01:30
DEV Beginners
查看原文

从零开始,学会安装Python、VS Code和Hugging Face库,下载并运行你的第一个AI模型。

准备工作:安装Python和代码编辑器

要运行AI模型,首先需要Python(一种编程语言,AI开发最常用)。去 python.org 下载最新版,安装时记得勾选“Add Python to PATH”。然后安装VS Code(免费代码编辑器),在官网下载安装即可。

安装AI库:Hugging Face Transformers

  1. 打开终端(Windows按Win+R输入cmd,Mac打开“终端”)。
  2. 输入以下命令安装Transformers库(它让你轻松使用成千上万的AI模型):
    pip install transformers
  3. 等待安装完成。如果遇到错误,试试先升级pip:python -m pip install --upgrade pip

下载并运行第一个AI模型

我们用Hugging Face上的一个文本生成模型来体验AI。在VS Code里新建文件,粘贴以下代码:

Tutorial Image
from transformers import pipeline
generator = pipeline('text-generation', model='gpt2')
result = generator('Once upon a time', max_length=50)
print(result)

保存为 test.py,然后在终端运行:python test.py。第一次运行会自动下载模型(约500MB),耐心等待。成功后会输出一段续写的文本。

验证是否成功

  • 如果看到类似 [{'generated_text': 'Once upon a time there was...'}] 的输出,说明环境搭建成功!
  • 如果报错,检查网络连接,或尝试更换模型(如 distilgpt2 更小更快)。

下一步可以做什么

现在你可以尝试其他模型:比如图片分类、翻译、问答。只需修改 pipeline 的任务名称和模型名称。记住:Hugging Face(一个AI模型共享平台)上有超过10万个模型,全部免费使用。

内容来源

DEV Beginners

发布时间

2026-06-03 01:30

返回 AI技术