PyTorch 深度学习笔记（一）PyTorch 入门——深度学习框架的选择与安装
系列导读：本系列共 6 篇，从 PyTorch 框架入门到实战案例，带你系统掌握深度学习开发。
一、什么是 PyTorch
PyTorch 是基于 Python 的深度学习框架，核心数据结构为张量（Tensor），支持动态计算图、自动微分、GPU 加速，广泛用于计算机视觉、自然语言处理、强化学习、大模型训练等领域。
1.1 核心定位
- 基于 Python：将数据封装成张量（Tensor）进行处理
- 灵活高效：提供构建、训练和部署机器学习和深度学习模型的完整工具链
- 广泛应用：学术研究 + 工业界，覆盖 CV、NLP、强化学习等全领域
1.2 核心组件速览
| 组件 | 作用 | 说明 |
|---|---|---|
| 张量 Tensor | 多维数组 | 支持 CPU/GPU 切换，兼容 NumPy |
| 自动微分 autograd | 自动计算梯度 | 支撑反向传播，无需手动求导 |
| torch.nn | 神经网络层 | 损失函数、基础模型基类 |
| torch.optim | 优化器 | SGD、Adam 等常用优化算法 |
| 数据工具 | DataLoader | 数据加载与批量处理 |
1.3 核心特性一览
- 动态图：运行时构建、修改结构，调试直观
- GPU 加速：一键切换 CUDA，训练速度飞跃
- 跨平台：Linux、Windows、macOS 全支持
- API 贴近 Python：简洁、一致、少魔法
- 易调试：像写普通 Python 代码一样调试神经网络
二、主流深度学习框架对比
2.1 双雄格局
当前深度学习框架领域呈现TensorFlow vs PyTorch双雄争霸格局，此外还有：
- MindSpore（华为）：国产 AI 框架，面向昇腾芯片优化
- PaddlePaddle（百度）：中文生态完善，产业级应用广泛
- Caffe（伯克利）：C++ 实现，速度极快但灵活性稍弱
2.2 PyTorch vs TensorFlow 详细对比
| 维度 | PyTorch | TensorFlow |
|---|---|---|
| 计算图 | 动态图（运行时构建） | TF1.x 静态图 / TF2.x 支持动态 |
| 调试难度 | 简单，像 Python | TF1.x 复杂，TF2.x 改善 |
| 上手难度 | 低，Pythonic | TF1.x 高，TF2.x 中等 |
| 科研生态 | 绝对优势，论文首选 | 工业部署生态更强 |
| 灵活性 | 极高，支持动态结构 | 相对固定 |
| 推荐场景 | 科研、原型、学习 | 大规模生产部署 |
2.3 选型建议
- 科研 / 学习 / 原型：优先选 PyTorch
- 大规模工业部署：TensorFlow 生态更成熟
- 国产芯片环境：MindSpore 适配更佳
三、PyTorch 发展简史
3.1 从 Torch 到 PyTorch
Torch（2002）
- 最早的 Torch 框架，由 Ronan Collobert、Clement Farabet 等人开发
- 科学计算框架，提供多维张量操作和科学计算工具
Torch7
- 引入 Lua 编程语言，在深度学习领域取得一定成功
- 随着 PyTorch 的普及，Torch 不再维护，Torch7 成为最后一个版本
PyTorch 0.1.0（2016）
- Facebook 人工智能研究院（FAIR）发布
- 标志着 PyTorch 正式诞生，迅速成为学术界主流
PyTorch 今日
- 学术论文、开源模型绝大多数使用 PyTorch
- 大模型、CV、NLP、强化学习全覆盖
- 社区活跃，生态完善
四、环境安装与部署
4.1 PyTorch 安装
# 基础安装（CPU 版本）pipinstalltorch-ihttps://pypi.tuna.tsinghua.edu.cn/simple# 查看已安装包pip list# 查看 torch 详情pip show torch4.2 验证安装
importtorch# 查看版本print(torch.__version__)# 检查 CUDA 是否可用print(torch.cuda.is_available())# True 表示 GPU 可用# 创建第一个张量x=torch.tensor([1,2,3])print(x)4.3 运行环境支持
| 维度 | 支持情况 |
|---|---|
| 硬件 | CPU、NVIDIA GPU（CUDA）、国产 AI 芯片、TPU |
| 系统 | Linux、Windows、macOS |
| 部署端 | 云端训练、边缘设备部署、移动端适配 |
五、下一篇预告
PyTorch 深度学习笔记（二）：PyTorch 优势全解析——为什么科研和工业都爱它
将深入剖析 PyTorch 的四大核心优势：动态图研究友好、张量+GPU+自动微分三合一、生态完善社区活跃、API 简洁上手快。
附录：推荐阅读
- PyTorch 官方文档
- PyTorch 官方教程
- Deep Learning with PyTorch 书籍
如果这篇文章对你有帮助，欢迎点赞、收藏、关注！你的支持是我持续创作的动力。