AI入门:手把手教你装好环境并跑通第一个模型
从零开始,学会安装Python、VS Code和Hugging Face库,下载并运行你的第一个AI模型。
准备工作:安装Python和代码编辑器
要运行AI模型,首先需要Python(一种编程语言,AI开发最常用)。去 python.org 下载最新版,安装时记得勾选“Add Python to PATH”。然后安装VS Code(免费代码编辑器),在官网下载安装即可。
安装AI库:Hugging Face Transformers
- 打开终端(Windows按Win+R输入cmd,Mac打开“终端”)。
- 输入以下命令安装Transformers库(它让你轻松使用成千上万的AI模型):
pip install transformers - 等待安装完成。如果遇到错误,试试先升级pip:
python -m pip install --upgrade pip
下载并运行第一个AI模型
我们用Hugging Face上的一个文本生成模型来体验AI。在VS Code里新建文件,粘贴以下代码:
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