d2l-pytorch
d2l-pytorch 是经典深度学习教材《动手学深度学习》(Dive Into Deep Learning)的 PyTorch 代码实现版本。该项目将原书基于 MXNet 框架的代码示例完整迁移至目前更主流的 PyTorch 框架,涵盖了从环境安装、数学基础、线性神经网络到多层感知机等核心章节的可运行代码笔记。
它主要解决了学习者在使用这本优质开源教材时面临的框架适配痛点。由于原版代码依赖 MXNet,而许多开发者和研究者更习惯或必须使用 PyTorch 生态,d2l-pytorch 填补了这一空白,让用户无需手动转换代码即可直接利用 PyTorch 进行复现和实验,极大地降低了学习门槛和时间成本。
这款资源非常适合希望系统学习深度学习的开发者、高校学生以及研究人员。无论是想要从零开始掌握神经网络原理的初学者,还是需要在 PyTorch 环境下快速验证算法的研究者,都能从中获益。其独特的技术亮点在于保留了原书“从零实现”与“简洁实现”并行的教学特色,既帮助读者深入理解底层数学逻辑,又展示了工业级的高效调用方式,是连接理论与实践的桥梁。需要注意的是,目前官方已推出完整的 PyTorch 移植版,建议用户结合最新官方资源共同参考学习。
使用场景
某高校人工智能实验室的研究生团队正致力于复现经典深度学习算法,以完成课程作业并夯实理论基础,但团队成员主要熟悉 PyTorch 框架。
没有 d2l-pytorch 时
- 框架迁移成本高:经典的《动手学深度学习》原书代码基于 MXNet 编写,学生需手动逐行将代码转换为 PyTorch 语法,极易引入难以排查的 Bug。
- 数学原理与代码脱节:在从零实现线性回归或多层感知机时,缺乏对应的 PyTorch 参考实现,导致学生难以验证自己对反向传播等核心数学推导的理解是否正确。
- 环境配置混乱:不同章节依赖的数据预处理和可视化工具分散,新手在搭建实验环境时往往因版本冲突或缺失依赖而耗费数天时间。
- 学习曲线陡峭:需要在理解深度学习理论的同时还要克服框架差异,导致注意力分散,严重拖慢了从理论到实践的转化效率。
使用 d2l-pytorch 后
- 即拿即用的 PyTorch 代码:直接获取书中所有章节(从数据操作到卷积神经网络)已完美适配 PyTorch 的代码,消除了手动移植的工作量和错误风险。
- 理论与实现无缝对照:通过“从零实现”与“简洁实现”两种版本的对比笔记,学生能清晰看到数学公式如何映射为具体的 PyTorch API,深刻掌握模型底层逻辑。
- 标准化的实验环境:项目提供了统一的安装指南和依赖管理,确保团队成员能在几分钟内复现书中的 Fashion-MNIST 分类等经典实验。
- 专注核心算法学习:移除了框架转换的干扰,团队能将全部精力集中在模型调优、过拟合分析及超参数选择等核心深度学习任务上。
d2l-pytorch 通过提供标准化的 PyTorch 版教学代码,彻底打通了深度学习理论学习与工业级框架实践之间的最后一公里。
运行环境要求
- 未说明
非必需(第 7.6 章涉及 GPU 使用,但未指定具体型号、显存或 CUDA 版本要求)
未说明

快速开始
更新:完整 PyTorch 版本请参阅原始仓库。我们不再维护此仓库。
本项目改编自 Aston Zhang、Zachary C. Lipton、Mu Li、Alex J. Smola 及全体社区贡献者所著的原版《深入浅出深度学习》(Dive Into Deep Learning)。原书 GitHub 地址:https://github.com/d2l-ai/d2l-en。我们致力于对本书进行修改,并将其中的 MXNet 代码片段转换为 PyTorch 格式。
注意:部分 ipynb 笔记本在 GitHub 上可能无法完美渲染。建议您克隆该仓库,或使用 nbviewer 来查看这些笔记本。
章节
Ch02 安装
Ch03 导论
Ch04 基础知识:速成课程
Ch05 线性神经网络
- 5.1 线性回归
- 5.2 从零开始实现线性回归
- 5.3 线性回归的简洁实现
- 5.4 Softmax 回归
- 5.5 图像分类数据集(Fashion-MNIST)
- 5.6 从零开始实现 Softmax 回归
- 5.7 Softmax 回归的简洁实现
Ch06 多层感知机
- 6.1 多层感知机
- 6.2 从零开始实现多层感知机
- 6.3 多层感知机的简洁实现
- 6.4 模型选择、欠拟合与过拟合
- 6.5 权重衰减
- 6.6 Dropout
- 6.7 前向传播、反向传播与计算图
- 6.8 数值稳定性与参数初始化
- 6.9 环境因素考量
- 6.10 在 Kaggle 上预测房价
Ch07 深度学习计算
Ch08 卷积神经网络
第9章 现代卷积网络
9.2 使用模块的网络(VGG)
9.3 网络中的网络(NiN)
9.5 批量归一化
9.6 残差网络(ResNet)
9.7 密集连接网络(DenseNet)
第10章 循环神经网络
- 10.1 序列模型
- 10.2 语言模型
- 10.3 循环神经网络
- 10.4 文本预处理
- 10.5 从零开始实现循环神经网络
- 10.6 循环神经网络的简洁实现
- 10.7 时间反向传播
- 10.8 门控循环单元(GRU)
- 10.9 长短期记忆网络(LSTM)
- 10.10 深度循环神经网络
- 10.11 双向循环神经网络
- 10.12 机器翻译与数据集
- 10.13 编码器-解码器架构
- 10.14 序列到序列
- 10.15 束搜索
第11章 注意力机制
- 11.1 注意力机制
- 11.2 带有注意力机制的序列到序列模型
- 11.3 Transformer
第12章 优化算法
第14章 计算机视觉
- 14.1 图像增强
- 14.2 微调
- 14.3 目标检测与边界框
- 14.4 锚框
- 14.5 多尺度目标检测
- 14.6 目标检测数据集(皮卡丘)
- 14.7 单次多盒检测(SSD)
- 14.8 基于区域的CNN(R-CNN)
- 14.9 语义分割与数据集
- 14.10 转置卷积
- 14.11 全卷积网络(FCN)
- 14.12 神经风格迁移
- 14.13 Kaggle上的图像分类(CIFAR-10)
- 14.14 Kaggle上的犬种识别(ImageNet Dogs)
贡献说明
欢迎随时提交Pull Request,为剩余章节贡献PyTorch笔记本。在开始编写笔记本之前,请先创建一个包含笔记本名称的议题,以便我们为您分配该任务(如果尚未有人被分配)。
请严格遵循IPython笔记本和子节的命名规范。
如果您认为某些部分需要更详细或更好的解释,请通过议题跟踪器提交问题,我们会尽快回复。
找到需要改进的代码并提交拉取请求。
发现我们遗漏的参考文献并提交拉取请求。
尽量避免提交过大的拉取请求,因为这会使理解和合并变得困难。建议分多次提交较小的请求。
支持
如果你喜欢这个仓库并觉得它有用,请考虑给它点个赞(★),这样它就能触达更广泛的受众了。
参考文献
[1] 原书 Dive Into Deep Learning -> GitHub 仓库
[2] 深度学习——直截了当
引用
如果你在研究中使用了本项目或代码,请使用以下 BibTeX 条目引用原书。
@book{zhang2020dive,
title={Dive into Deep Learning},
author={Aston Zhang and Zachary C. Lipton and Mu Li and Alexander J. Smola},
note={\url{https://d2l.ai}},
year={2020}
}
常见问题
相似工具推荐
openclaw
OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你
stable-diffusion-webui
stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。
everything-claude-code
everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上
ComfyUI
ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。
gemini-cli
gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。
markitdown
MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器