极客前沿

AI入门:一步步安装并跑通你的第一个模型

2026-05-20 01:30
DEV Tutorial
查看原文

本文教你从零开始安装AI开发环境,并运行第一个简单模型,适合完全没接触过AI的初学者。

准备环境:你需要装什么

在开始之前,先了解两个基本概念:Python 是一种编程语言,AI 开发最常用它;PyTorch 是一个工具包,帮你快速搭建和运行 AI 模型。如果你完全没接触过编程,别担心,跟着下面步骤走就行。

  • 下载 Python:访问 python.org,点击“Download”按钮,选择最新版(比如 3.12)。安装时记得勾选“Add Python to PATH”。
  • 打开命令行:Windows 用户按 Win+R,输入 cmd 回车;Mac 用户打开“终端”。
  • 验证 Python 是否装好:在命令行输入 python --version,如果显示版本号,说明成功。

安装 PyTorch:一行命令搞定

PyTorch 是 Facebook 开发的深度学习框架,你可以把它理解成“AI 积木盒”。安装很简单:

Tutorial Image
  1. 打开命令行。
  2. 输入以下命令并按回车:pip install torch torchvision torchaudio
  3. 等待几分钟,看到“Successfully installed”就装好了。
  4. 常见坑:如果提示“pip 不是命令”,说明 Python 没加到 PATH,请重装 Python 并勾选那个选项。

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

我们用一个经典例子——识别手写数字,来验证环境是否正常。这个模型叫 MNIST,就是一堆 0~9 的手写图片数据集。

  1. 在桌面新建一个文本文件,命名为 first_model.py
  2. 用记事本打开,复制以下代码:
import torch
import torchvision
import torchvision.transforms as transforms

# 下载数据集(第一次需要联网)
train_set = torchvision.datasets.MNIST(root='./data', train=True, download=True, transform=transforms.ToTensor())
print(f"数据集大小:{len(train_set)} 张图片")
print("环境配置成功!")
  1. 保存文件,在命令行进入桌面(输入 cd Desktop),然后运行 python first_model.py
  2. 如果看到“数据集大小:60000 张图片”和“环境配置成功!”,恭喜你,AI 环境已经跑通了!

下一步可以做什么

  • 学习基础 Python 语法(比如变量、循环),能帮你理解更多 AI 代码。
  • 尝试运行官方教程中的简单模型,比如用 PyTorch 训练一个图片分类器。
  • 加入社区:遇到问题可以搜索“PyTorch 入门教程”或去论坛提问。

内容来源

DEV Tutorial

发布时间

2026-05-20 01:30

返回 AI技术