极客前沿

一步步搭建AI开发环境:从零开始跑通第一个模型

2026-06-11 01:31
DEV Tutorial
查看原文

从零开始,学会下载安装Python、CUDA和PyTorch,并跑通第一个图像识别模型。

准备环境

首先,你需要安装Python(一种编程语言,用来写AI代码)。去官网python.org下载最新版,安装时记得勾选“Add Python to PATH”。然后打开终端(Windows搜cmd,Mac搜终端),输入python --version,如果显示版本号就成功了。

安装AI框架

接下来安装PyTorch(一个AI工具包,帮你快速搭建神经网络)。打开终端,输入以下命令:pip install torch torchvision。如果电脑有NVIDIA显卡,想用GPU加速,先去NVIDIA官网下载CUDA(一个加速工具),再运行pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118

跑通第一个模型

新建一个Python文件,比如first_ai.py,复制以下代码:

Tutorial Image
import torch
from torchvision import models, transforms
from PIL import Image

# 加载预训练的ResNet模型(一个图像识别模型)
model = models.resnet18(pretrained=True)
model.eval()

# 处理图片
img = Image.open('cat.jpg')  # 替换成你自己的图片
transform = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
])
img_tensor = transform(img).unsqueeze(0)

# 预测
with torch.no_grad():
    output = model(img_tensor)
print(output.argmax())  # 输出类别编号

运行python first_ai.py,如果输出一个数字(比如285对应“埃及猫”),就说明跑通了!

常见坑与下一步

常见坑: 1. 图片路径错误:确保图片和脚本在同一文件夹。2. 缺少库:用pip install pillow安装PIL。3. GPU版本装错:先查CUDA版本,用nvidia-smi查看。

下一步: 尝试修改代码识别自己的照片,或者学习用Jupyter Notebook交互式编程。恭喜你,已经迈出了AI第一步!

内容来源

DEV Tutorial

发布时间

2026-06-11 01:31

返回 AI技术