极客前沿

AI 入门:手把手装环境、跑通第一个模型

2026-06-26 01:31
DEV Tutorial
查看原文

从零开始,学会安装 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 版,也能跑。

Tutorial Image
  1. 打开 PyTorch 官网 pytorch.org,点击“Get Started”。
  2. 选择你的系统、包管理器(推荐 pip)、Python 版本,以及 CUDA(有 GPU 选对应版本,没有就选 None)。
  3. 复制生成的安装命令(例如 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

返回 AI技术