Dive-into-DL-PyTorch

GitHub
19.4k 5.4k 简单 1 次阅读 今天Apache-2.0图像开发框架语言模型
AI 解读 由 AI 自动生成,仅供参考

Dive-into-DL-PyTorch 是经典教材《动手学深度学习》的 PyTorch 版本重构项目。原书代码基于 MXNet 框架,而本项目将其完整迁移至目前业界更主流的 PyTorch 生态,旨在降低学习门槛,让读者能直接使用熟悉的工具掌握深度学习核心知识。

该项目主要解决了学习者面对优质中文教材却受限于非主流框架(MXNet)的痛点。它保留了原书从数学原理到代码实现的完整教学体系,涵盖线性回归、卷积神经网络等基础内容,同时提供了基于 PyTorch 的可运行 Jupyter Notebook 代码和配套文档。无论是环境配置、数据操作还是模型训练,所有示例均已完成适配,确保理论与实践无缝衔接。

Dive-into-DL-PyTorch 特别适合希望入门深度学习的学生、开发者及研究人员。使用者无需具备深厚的机器学习背景,只要掌握基础的线性代数、微积分知识及 Python 编程能力即可上手。项目的一大亮点是其灵活的文档阅读体验:除了直接克隆代码运行外,还利用 Docsify 搭建了在线网页版,并支持通过本地服务或 Docker 容器快速部署文档环境,方便用户在不同场景下高效学习与查阅。这是一个开源社区驱动的优质资源,欢迎各界人士共同参与贡献。

使用场景

计算机专业大三学生李明正在准备深度学习课程的大作业,他需要快速掌握卷积神经网络(CNN)的原理并复现图像分类模型,但手头只有基于 MXNet 框架的经典教材《动手学深度学习》。

没有 Dive-into-DL-PyTorch 时

  • 框架转换困难:教材代码全是 MXNet 写法,而实验室服务器和主流社区资源均基于 PyTorch,李明需手动逐行翻译代码,极易出错。
  • 环境配置混乱:为了运行原书代码,不得不额外安装冷门的 MXNet 环境,导致与现有的 PyTorch 项目依赖冲突,调试耗时巨大。
  • 理论实践脱节:在理解反向传播或自定义层等复杂概念时,无法直接运行对应的 PyTorch 示例进行验证,只能靠空想推导,学习曲线陡峭。
  • 资源查找低效:遇到报错时,无法直接将教材代码片段放入 PyTorch 社区搜索解决方案,因为语法体系完全不同,排查问题如大海捞针。

使用 Dive-into-DL-PyTorch 后

  • 代码无缝衔接:直接获取书中所有章节的 PyTorch 原生实现,李明可立即在现有环境中运行线性回归、Softmax 及 CNN 等核心算法,无需手动转译。
  • 环境统一高效:无需安装额外的 MXNet,完全复用现有的 PyTorch 开发栈,通过 Jupyter Notebook 即可边看文档边跑代码,环境零冲突。
  • 即时验证原理:针对“多输入通道”或“丢弃法”等难点,直接修改提供的 PyTorch 代码参数观察输出变化,将抽象数学公式转化为直观的张量运算结果。
  • 调试事半功倍:遇到的任何报错均可直接在 PyTorch 生态中检索,且代码结构与主流开源项目一致,快速定位并解决了模型不收敛的问题。

Dive-into-DL-PyTorch 消除了经典教材与现代主流框架之间的语言隔阂,让学习者能专注于算法逻辑本身而非繁琐的代码移植。

运行环境要求

操作系统
  • 未说明
GPU

非必需(书中包含 GPU 计算章节,但基础学习可在 CPU 运行),具体型号及 CUDA 版本未说明

内存

未说明

依赖
notes本项目是《动手学深度学习》的 PyTorch 实现版本。主要运行内容为 Jupyter Notebook 代码。若需本地浏览文档,可选择安装 Node.js 和 docsify-cli,或使用 Docker 容器运行网页服务。书中部分章节涉及 GPU 加速和多 GPU 计算,建议有相关硬件以获得完整体验,但未强制要求。
python未说明
PyTorch
docsify-cli (可选,用于本地文档服务)
Docker (可选,用于容器化部署)
Dive-into-DL-PyTorch hero image

快速开始

封面

本项目《动手学深度学习》 原书中MXNet代码实现改为PyTorch实现。原书作者:阿斯顿·张、李沐、扎卡里 C. 立顿、亚历山大 J. 斯莫拉以及其他社区贡献者,GitHub地址:https://github.com/d2l-ai/d2l-zh

此书的版本存在一些不同,针对此书英文版的PyTorch重构可参考这个项目。 这本书的中文版英文版之间存在一些差异。若需对英文版进行PyTorch重构,可参考该项目

简介

本仓库主要包含code和docs两个文件夹(外加一些数据存放在data中)。其中code文件夹就是每章相关jupyter notebook代码(基于PyTorch);docs文件夹就是markdown格式的《动手学深度学习》书中的相关内容,然后利用docsify将网页文档部署到GitHub Pages上,由于原书使用的是MXNet框架,所以docs内容可能与原书略有不同,但是整体内容是一样的。欢迎对本项目做出贡献或提出issue。

面向人群

本项目面向对深度学习感兴趣,尤其是想使用PyTorch进行深度学习的童鞋。本项目并不要求你有任何深度学习或者机器学习的背景知识,你只需了解基础的数学和编程,如基础的线性代数、微分和概率,以及基础的Python编程。

食用方法

方法一

本仓库包含一些latex公式,但github的markdown原生是不支持公式显示的,而docs文件夹已经利用docsify被部署到了GitHub Pages上,所以查看文档最简便的方法就是直接访问本项目网页版。当然如果你还想跑一下运行相关代码的话还是得把本项目clone下来,然后运行code文件夹下相关代码。

方法二

你还可以在本地访问文档,先安装docsify-cli工具:

npm i docsify-cli -g

然后将本项目clone到本地:

git clone https://github.com/ShusenTang/Dive-into-DL-PyTorch.git
cd Dive-into-DL-PyTorch

然后运行一个本地服务器,这样就可以很方便的在http://localhost:3000实时访问文档网页渲染效果。

docsify serve docs

方法三

如果你不想安装docsify-cli工具,甚至你的电脑上都没有安装Node.js,而出于某些原因你又想在本地浏览文档,那么你可以在docker容器中运行网页服务。

首先将本项目clone到本地:

git clone https://github.com/ShusenTang/Dive-into-DL-PyTorch.git
cd Dive-into-DL-PyTorch

之后使用如下命令创建一个名称为「d2dl」的docker镜像:

docker build -t d2dl .

镜像创建好后,运行如下命令创建一个新的容器:

docker run -dp 3000:3000 d2dl

最后在浏览器中打开这个地址http://localhost:3000/#/,就能愉快地访问文档了。适合那些不想在电脑上装太多工具的小伙伴。

目录

持续更新中......

原书地址

中文版:动手学深度学习 | Github仓库
English Version: Dive into Deep Learning | Github Repo

引用

如果您在研究中使用了这个项目请引用原书:

@book{zhang2019dive,
    title={Dive into Deep Learning},
    author={Aston Zhang and Zachary C. Lipton and Mu Li and Alexander J. Smola},
    note={\url{http://www.d2l.ai}},
    year={2020}
}

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

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 真正成长为懂上

159.3k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

108.3k|★★☆☆☆|1周前
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

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 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架