AI入门:一步步安装并跑通你的第一个模型
本文教你从零开始安装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 积木盒”。安装很简单:
- 打开命令行。
- 输入以下命令并按回车:
pip install torch torchvision torchaudio - 等待几分钟,看到“Successfully installed”就装好了。
- 常见坑:如果提示“pip 不是命令”,说明 Python 没加到 PATH,请重装 Python 并勾选那个选项。
跑通第一个模型:识别手写数字
我们用一个经典例子——识别手写数字,来验证环境是否正常。这个模型叫 MNIST,就是一堆 0~9 的手写图片数据集。
- 在桌面新建一个文本文件,命名为
first_model.py。 - 用记事本打开,复制以下代码:
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("环境配置成功!")
- 保存文件,在命令行进入桌面(输入
cd Desktop),然后运行python first_model.py。 - 如果看到“数据集大小:60000 张图片”和“环境配置成功!”,恭喜你,AI 环境已经跑通了!
下一步可以做什么
- 学习基础 Python 语法(比如变量、循环),能帮你理解更多 AI 代码。
- 尝试运行官方教程中的简单模型,比如用 PyTorch 训练一个图片分类器。
- 加入社区:遇到问题可以搜索“PyTorch 入门教程”或去论坛提问。
内容来源
DEV Tutorial
发布时间
2026-05-20 01:30