AI入门实战:一步步安装并跑通你的第一个AI模型
零基础学会下载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工具,特别好上手。打开终端,输入以下命令:
- 先升级pip(Python的包管理器):
python -m pip install --upgrade pip - 安装PyTorch:
pip install torch torchvision torchaudio - 等待下载完成,大概几分钟。如果慢可以换国内镜像,比如
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision torchaudio
常见坑:如果提示“pip不是内部命令”,说明Python没加到PATH,重装时勾选那个选项即可。
第三步:跑通第一个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