AI 入门:手把手教你安装与运行第一个模型
零基础读者将学会搭建 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 模型,无需自己训练。在终端中运行以下命令:
- 输入
pip install transformers并回车,等待安装完成。 - 再安装 torch(PyTorch,一个深度学习框架),输入
pip install torch。如果网速慢,可以加国内镜像,例如pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple。
常见坑:如果报错“pip 不是内部或外部命令”,说明 Python 未正确添加到环境变量,重新安装并勾选“Add to PATH”即可。
运行第一个模型
创建一个新文件,命名为 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