极客前沿

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

2026-07-03 01:32
DEV Machine Learning
查看原文

零基础学会下载Python、安装PyTorch、运行简单的AI示例,感受AI的实际效果。

很多人觉得AI是高大上的技术,其实你也能亲手跑通一个AI程序。本文带你从零开始,安装好环境,运行第一个AI模型,体验AI的魔力。

第一步:准备环境

你需要一台电脑(Windows、Mac或Linux都可以),然后安装Python。Python是AI最常用的编程语言,简单易学。去官网python.org下载最新版(比如3.11),安装时记得勾选“Add Python to PATH”。

  • 下载Python安装包
  • 运行安装程序,勾选“Add Python to PATH”
  • 安装完成后,打开终端(命令提示符或终端),输入python --version,看到版本号就成功了

第二步:安装AI框架

我们用一个叫PyTorch的库,它是Facebook开发的AI工具,特别好上手。打开终端,输入以下命令:

  1. 先升级pip(Python的包管理器):python -m pip install --upgrade pip
  2. 安装PyTorch:pip install torch torchvision torchaudio
  3. 等待下载完成,大概几分钟。如果慢可以换国内镜像,比如pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision torchaudio

常见坑:如果提示“pip不是内部命令”,说明Python没加到PATH,重装时勾选那个选项即可。

Tutorial Image

第三步:跑通第一个AI程序

AI最经典的任务是识别图片。我们让AI识别一张手写数字图片。新建一个文件test.py,用记事本打开,粘贴以下代码:

import torch
from torchvision import datasets, transforms

# 下载手写数字数据集(MNIST)
train_data = datasets.MNIST(root='./data', train=True, download=True, transform=transforms.ToTensor())

# 显示第一张图片
import matplotlib.pyplot as plt
plt.imshow(train_data[0][0].squeeze(), cmap='gray')
plt.show()

print('标签是:', train_data[0][1])

然后在终端运行:python test.py。如果弹出窗口显示一个数字图片,说明你成功跑通了AI!

解释:这段代码下载了MNIST数据集(包含6万张手写数字图片),然后显示第一张图片和它的标签(实际数字)。

第四步:下一步可以做什么

你已经入门了!接下来可以:

  • 尝试修改代码,识别你自己的图片
  • 学习用AI做预测(比如房价预测)
  • 了解更高级的模型,比如ChatGPT背后的Transformer

AI并不神秘,动手试试,你会发现它很有趣。

内容来源

DEV Machine Learning

发布时间

2026-07-03 01:32

返回 AI技术