极客前沿

AI 入门实战:装好环境,跑通第一个模型

2026-06-27 01:33
DEV Machine Learning
查看原文

本文带你从零开始安装 Python 和 PyTorch,跑通第一个 AI 模型,并理解 AI 与“生命”的根本区别。

准备环境:你需要什么?

要跑 AI 模型,你只需要一台电脑(Windows/Mac/Linux 都行)和一点点耐心。我们会用到两个核心工具:

  • Python:一种简单易学的编程语言,AI 领域几乎都用它。去 python.org 下载安装包,记得勾选“Add Python to PATH”。
  • PyTorch:一个 AI 框架,帮你快速搭建和运行模型。安装命令见下一步。

常见坑:安装 Python 时一定要勾选“Add Python to PATH”,否则系统找不到 Python 命令。

Tutorial Image

安装步骤:三步搞定

  1. 打开终端(Windows 搜索“cmd”,Mac/Linux 搜索“终端”)。
  2. 安装 PyTorch:复制下面这条命令并回车运行(它会自动安装 PyTorch 和依赖)
    pip install torch torchvision torchaudio
  3. 验证安装:在终端输入 python -c "import torch; print(torch.__version__)",如果看到版本号(如 2.0.0),说明安装成功。

常见坑:如果安装很慢,可以加国内镜像,例如 pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

跑通第一个模型:识别手写数字

下面这段代码会加载一个已经训练好的模型,识别图片里的手写数字(0-9)。复制到 Python 文件(比如 test.py)中运行:

Tutorial Image
import torch
import torchvision.transforms as transforms
from PIL import Image

# 加载预训练模型
model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=True)
model.eval()

# 处理图片
transform = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
])
img = Image.open('your_image.jpg')
img_t = transform(img)
batch_t = torch.unsqueeze(img_t, 0)

# 预测
out = model(batch_t)
_, index = torch.max(out, 1)
print('预测类别编号:', index.item())

your_image.jpg 换成你本地的图片路径,运行后就会输出一个数字,代表模型认为图片属于哪个类别。

常见坑:如果报错缺少 PIL,先运行 pip install pillow

理解背后的道理:AI 和“生命”的区别

跑通模型后,你可能会想:AI 真的“懂”吗?其实,现在的 AI 更像一个超级模仿者——它能描述世界,但没有自己的“生命”。就像原文说的:一个飓风模型不会让你淋湿,模拟火焰不会提高室温。AI 预测了每一个细胞的战斗,但它从未参与其中。这就是为什么顶尖科学家 Yann LeCun 要花十亿美元研究“世界模型”——让 AI 理解因果关系,而不只是预测下一个词。

不过,作为新手,你只需要先享受跑通第一个模型的成就感。下一步可以试试修改代码,看看不同图片的结果,或者去 PyTorch 官网找教程深入学习。

内容来源

DEV Machine Learning

发布时间

2026-06-27 01:33

返回 AI技术