极客前沿

AI入门:从零搭建环境,第一次跑通深度学习模型

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

手把手教你安装Python、配置环境、运行第一个简单的AI模型,零基础也能轻松上手。

准备工作:安装Python和工具包

首先需要安装Python(一种编程语言,AI模型通常用它编写)。去官网 python.org 下载最新版(建议3.8以上),安装时记得勾选“Add Python to PATH”。

接着安装pip(Python的包管理器,用来安装其他工具)。通常Python自带pip,在命令行输入 pip --version 检查是否安装成功。

安装深度学习框架:PyTorch

PyTorch(一个流行的深度学习框架,帮你快速搭建和训练AI模型)是入门首选。打开命令行(Windows按Win+R输入cmd,Mac打开终端),输入以下命令安装CPU版本(没有独立显卡也能用):

  • pip install torch torchvision torchaudio

如果你有NVIDIA显卡并想用GPU加速,先安装CUDA(英伟达的并行计算平台,让显卡跑AI),然后去PyTorch官网 pytorch.org 选择对应命令安装。

Tutorial Image

验证是否成功:跑一个简单模型

创建一个Python文件(比如 test.py),写入以下代码:

import torch
x = torch.tensor([1.0, 2.0, 3.0])
print(x)
print('PyTorch版本:', torch.__version__)

在命令行运行 python test.py,如果输出类似 tensor([1., 2., 3.]) 和版本号,说明环境配置成功!

下一步可以做什么

你已经能跑PyTorch了!接下来可以:

  • 学习用MNIST数据集(手写数字图片,AI入门常用)训练一个识别数字的模型
  • 尝试在线教程(比如官方60分钟入门)
  • 探索Jupyter Notebook(交互式编程环境,方便边写边看结果)

遇到问题别怕,常见坑:Python版本太低、pip未更新(运行 pip install --upgrade pip)、路径有中文。慢慢来,你已经在路上了!

内容来源

DEV Machine Learning

发布时间

2026-07-03 01:32

返回 AI技术