AI 入门:手把手装环境、跑通第一个模型
从零开始,学会安装 Python、PyTorch,并运行一个简单的 AI 模型,适合完全没接触过 AI 的新手。
很多人觉得 AI 很神秘,其实入门并不难。这篇教程带你从安装环境开始,一步步跑通你的第一个 AI 程序。你只需要一台能上网的电脑,跟着做就行。
准备环境:装 Python 和必备工具
AI 开发最常用的语言是 Python(一种简单易学的编程语言)。首先,确保你的电脑上安装了 Python 3.8 或更高版本。
- Windows 用户:去 python.org 下载安装包,安装时记得勾选“Add Python to PATH”。
- Mac 用户:打开终端,输入
brew install python3(需先安装 Homebrew)。 - Linux 用户:终端输入
sudo apt install python3 python3-pip(Ubuntu/Debian)。
安装后,打开终端(命令提示符),输入 python3 --version 验证是否成功。如果显示版本号,就对了。
安装 AI 框架:PyTorch
PyTorch 是一个流行的 AI 框架,帮我们快速搭建和训练模型。安装前,先确认你的电脑有没有独立显卡(NVIDIA GPU)。如果有,可以装 GPU 版(更快);没有就装 CPU 版,也能跑。
- 打开 PyTorch 官网 pytorch.org,点击“Get Started”。
- 选择你的系统、包管理器(推荐 pip)、Python 版本,以及 CUDA(有 GPU 选对应版本,没有就选 None)。
- 复制生成的安装命令(例如
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu),粘贴到终端运行。
等待下载安装。如果网速慢,可以加国内镜像源,比如在命令后加 -i https://pypi.tuna.tsinghua.edu.cn/simple。
跑通第一个模型:图像分类
我们用一个预训练好的 ResNet-18 模型(一种能识别图片内容的 AI 模型)来测试环境。新建一个 Python 文件 test.py,写入以下代码:
import torch
from torchvision import models, transforms
from PIL import Image
import requests
# 加载预训练模型
model = models.resnet18(pretrained=True)
model.eval()
# 下载一张测试图片
url = "https://upload.wikimedia.org/wikipedia/commons/thumb/4/4d/Cat_August_2010-4.jpg/1280px-Cat_August_2010-4.jpg"
img = Image.open(requests.get(url, stream=True).raw)
# 预处理图片
transform = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
img_tensor = transform(img).unsqueeze(0)
# 预测
with torch.no_grad():
output = model(img_tensor)
# 读取类别标签
with open("imagenet_classes.txt", "w") as f:
f.write(requests.get("https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt").text)
with open("imagenet_classes.txt", "r") as f:
labels = [line.strip() for line in f.readlines()]
_, predicted_idx = torch.max(output, 1)
print("预测结果:", labels[predicted_idx[0]])在终端运行 python test.py,如果输出“预测结果: tabby cat”之类的文字,说明环境配置成功,你已经跑通了第一个 AI 模型!
常见问题与下一步
- 报错“ModuleNotFoundError: No module named 'torch'”:说明 PyTorch 没装好,重新运行安装命令。
- 下载图片失败:检查网络,或换一张图片链接。
- 内存不足:可能是图片太大,换小一点的图片。
接下来,你可以尝试修改代码,识别你自己的图片,或者学习如何训练新模型。祝你在 AI 世界里玩得开心!
内容来源
DEV Tutorial
发布时间
2026-06-26 01:31