AI入门:从零搭建环境,第一次跑通深度学习模型
手把手教你安装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 选择对应命令安装。
验证是否成功:跑一个简单模型
创建一个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