PyTorch 深度学习笔记(一)PyTorch 入门——深度学习框架的选择与安装

📅 2026/7/31 3:11:59
PyTorch 深度学习笔记(一)PyTorch 入门——深度学习框架的选择与安装
系列导读本系列共 6 篇从 PyTorch 框架入门到实战案例带你系统掌握深度学习开发。一、什么是 PyTorchPyTorch 是基于 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 详细对比维度PyTorchTensorFlow计算图动态图运行时构建TF1.x 静态图 / TF2.x 支持动态调试难度简单像 PythonTF1.x 复杂TF2.x 改善上手难度低PythonicTF1.x 高TF2.x 中等科研生态绝对优势论文首选工业部署生态更强灵活性极高支持动态结构相对固定推荐场景科研、原型、学习大规模生产部署2.3 选型建议科研 / 学习 / 原型优先选 PyTorch大规模工业部署TensorFlow 生态更成熟国产芯片环境MindSpore 适配更佳三、PyTorch 发展简史3.1 从 Torch 到 PyTorchTorch2002最早的 Torch 框架由 Ronan Collobert、Clement Farabet 等人开发科学计算框架提供多维张量操作和科学计算工具Torch7引入 Lua 编程语言在深度学习领域取得一定成功随着 PyTorch 的普及Torch 不再维护Torch7 成为最后一个版本PyTorch 0.1.02016Facebook 人工智能研究院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 可用# 创建第一个张量xtorch.tensor([1,2,3])print(x)4.3 运行环境支持维度支持情况硬件CPU、NVIDIA GPUCUDA、国产 AI 芯片、TPU系统Linux、Windows、macOS部署端云端训练、边缘设备部署、移动端适配五、下一篇预告PyTorch 深度学习笔记二PyTorch 优势全解析——为什么科研和工业都爱它将深入剖析 PyTorch 的四大核心优势动态图研究友好、张量GPU自动微分三合一、生态完善社区活跃、API 简洁上手快。附录推荐阅读PyTorch 官方文档PyTorch 官方教程Deep Learning with PyTorch 书籍如果这篇文章对你有帮助欢迎点赞、收藏、关注你的支持是我持续创作的动力。