AI 入门:装好环境,跑通你的第一个模型
从零开始,教你安装 Python、PyTorch,并运行一个简单的 AI 图像生成程序。
准备环境:你需要装什么?
要跑 AI 程序,先装好 Python。Python 是 AI 最常用的编程语言,就像盖房子用的砖头。去 python.org 下载最新版,安装时记得勾选“Add Python to PATH”(把 Python 加到系统路径,这样你才能在命令行里直接调用它)。
安装步骤:一步步来
- 打开终端(Windows 按 Win+R 输入 cmd,Mac 打开“终端”)。
- 输入
python --version,看到版本号说明安装成功。 - 安装包管理器 pip(一般自带,可输入
pip --version检查)。 - 安装 PyTorch:去 pytorch.org 选你的系统,复制命令粘贴执行。PyTorch 是 AI 模型的核心库,相当于引擎。
- 安装 transformers:输入
pip install transformers。这个库让你方便调用预训练模型。
验证是否成功:跑第一个 AI 程序
创建一个新文件 test.py,用记事本打开,粘贴以下代码:
from transformers import pipeline
generator = pipeline('text-generation', model='gpt2')
result = generator('Once upon a time', max_length=50)
print(result[0]['generated_text'])保存后在终端运行 python test.py。如果看到一段英文故事,恭喜你——AI 环境装好了!
下一步可以做什么
- 换模型:把
gpt2换成distilgpt2或bert-base-uncased,体验不同效果。 - 试试图像生成:安装 diffusers 库(
pip install diffusers),用类似代码生成图片。 - 常见坑:如果安装慢,换国内镜像源(如清华源);如果报错缺库,用
pip install 库名补装。
内容来源
DEV Machine Learning
发布时间
2026-05-11 01:33