极客前沿

AI 入门:手把手教你安装与运行第一个模型

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

零基础读者将学会搭建 AI 环境、下载并运行一个简单的语言模型,完成第一次 AI 体验。

准备环境

首先,你需要一台能联网的电脑。我们使用 Python(一种简单易学的编程语言)来操作 AI 模型。如果你还没装 Python,请到 python.org 下载最新版(推荐 3.10 或更高),安装时记得勾选“Add Python to PATH”。

  • 安装完成后,打开命令提示符(Windows 搜索“cmd”)或终端(Mac/Linux),输入 python --version,如果显示版本号,说明安装成功。
  • 接着安装 pip(Python 的包管理器),通常 Python 自带,检查方法:输入 pip --version

安装 AI 库

我们用 Hugging Face Transformers 库来加载模型。这个库让你能轻松调用各种 AI 模型,无需自己训练。在终端中运行以下命令:

  1. 输入 pip install transformers 并回车,等待安装完成。
  2. 再安装 torch(PyTorch,一个深度学习框架),输入 pip install torch。如果网速慢,可以加国内镜像,例如 pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

常见坑:如果报错“pip 不是内部或外部命令”,说明 Python 未正确添加到环境变量,重新安装并勾选“Add to PATH”即可。

Tutorial Image

运行第一个模型

创建一个新文件,命名为 test_model.py,用记事本或任何编辑器打开,粘贴以下代码:

from transformers import pipeline

# 加载一个文本生成模型(GPT-2 的小版本)
pipe = pipeline("text-generation", model="distilgpt2")

# 输入一句话,让模型续写
result = pipe("今天天气真好,我想", max_length=20)
print(result[0]['generated_text'])

保存后,在终端运行 python test_model.py。第一次运行会下载模型(约 300MB),之后就能看到 AI 续写的句子。恭喜,你已跑通第一个 AI 模型!

下一步做什么

你可以尝试其他任务,比如情感分析(判断一句话是正面还是负面):

  • pipeline 的参数改为 "sentiment-analysis",输入“I love this!”试试。
  • 或者换成 中文模型,例如 model="uer/roberta-base-finetuned-cluener2020-chinese" 做命名实体识别。

遇到问题?先检查网络连接,或搜索错误信息。AI 学习之路刚刚开始,享受探索的乐趣吧!

内容来源

DEV Machine Learning

发布时间

2026-06-29 01:32

返回 AI技术