DRL-Pytorch
DRL-Pytorch 是一个基于 PyTorch 框架构建的深度强化学习(DRL)算法开源库,旨在提供一套干净、稳健且统一的代码实现。它涵盖了从经典的 Q-learning、Duel DDQN 到前沿的 PPO、TD3、SAC 以及分布式架构 ASL 等主流算法,全面支持离散与连续动作空间的任务需求。
在强化学习领域,研究者常面临不同算法代码风格迥异、依赖环境复杂难以复现等痛点。DRL-Pytorch 通过标准化的项目结构和清晰的模块设计,有效解决了这些问题。用户只需进入对应算法文件夹运行 main.py 即可从零开始训练,极大地降低了上手门槛和调试成本。此外,该库还贴心地提供了各算法的独立仓库链接,方便按需调用。
这套工具特别适合人工智能领域的研究人员、高校学生以及希望快速验证想法的开发者使用。无论是用于学术实验对比,还是作为学习深度强化学习原理的教学素材,DRL-Pytorch 都能提供可靠的支持。其技术亮点在于不仅兼容标准的 Gymnasium 仿真环境,还展示了与 NVIDIA Isaac Sim 等高性能物理引擎的集成潜力,为需要大规模并行训练或机器人控制的高级应用预留了广阔空间。如果你正在寻找一份代码规范、文档详尽的 DRL 实战参考,DRL-Pytorch 将是一个理想的选择。
使用场景
某自动驾驶初创公司的算法团队正致力于训练一台移动机器人,使其在复杂的动态仓库环境中实现自主避障与高效路径规划。
没有 DRL-Pytorch 时
- 重复造轮子效率低:团队成员需从零手写 PPO、TD3 等复杂算法的基础架构,耗费数周时间调试网络结构与环境交互接口,严重拖慢研发进度。
- 代码风格不统一:不同成员实现的算法代码风格迥异,数据预处理、奖励函数定义及模型保存方式各不相同,导致代码审查困难且难以复用。
- 复现经典算法难:尝试复现论文中的 C51 或 SAC 算法时,常因细节处理不当(如噪声注入、优先经验回放)导致模型无法收敛,排查问题如同大海捞针。
- 环境适配成本高:每次切换仿真环境(从 Gym 到 Isaac Sim),都需要大幅修改底层代码以适配新的观测空间与动作空间,缺乏统一的抽象层。
使用 DRL-Pytorch 后
- 开箱即用加速迭代:直接调用 DRL-Pytorch 中预置的 PPO 和 TD3 模块,仅需几行代码即可启动训练,将算法验证周期从数周缩短至几天。
- 统一标准便于协作:所有算法均采用一致的代码结构与接口规范,团队成员可轻松互换模块进行对比实验,显著提升了协作效率与代码可维护性。
- 鲁棒实现保障收敛:依托库内经过严格测试的 Duel DDQN、PER 等高级特性,模型在复杂场景下快速收敛,避免了因底层实现缺陷导致的训练失败。
- 灵活适配多类环境:凭借其对 Gymnasium 等标准环境的原生支持及清晰的接口设计,团队能迅速将算法迁移至自研的 Sparrow 机器人仿真器中,无需重构核心逻辑。
DRL-Pytorch 通过提供干净、健壮且统一的算法实现,让研发团队从繁琐的底层编码中解放出来,专注于解决实际的决策控制难题。
运行环境要求
- 未说明
- 未说明 (项目基于 PyTorch,支持 CPU 或 GPU 运行,具体取决于所选算法和环境复杂度
- 如使用 Isaac Sim 等高级仿真器则需 NVIDIA GPU)
未说明

快速开始
0.星标历史
1.依赖项
除非另有说明,本仓库使用以下Python依赖项:
gymnasium==0.29.1
numpy==1.26.1
pytorch==2.1.0
python==3.11.5
2.如何使用我的代码
进入您想要使用的算法文件夹,并运行main.py从头开始训练:
python main.py
更多详细信息,请查看相应算法文件夹中的README.md文件。
3.代码独立链接
- 1.Q-learning
- 2.1双duel DQN
- 2.2Atari游戏上的噪声duel DDQN
- 2.3优先级经验回放(PER) DQN/DDQN
- 2.4分类DQN (C51)
- 2.5噪声网络DQN
- 3.1离散动作空间的近端策略优化(PPO)
- 3.2连续动作空间的近端策略优化(PPO)
- 4.1深度确定性策略梯度(DDPG)
- 4.2延迟双深度确定性策略梯度(TD3)
- 5.1离散动作空间的软演员评论家(SAC)
- 5.2连续动作空间的软演员评论家(SAC)
- 6.演员-共享者-学习者(ASL)
4.深度强化学习推荐资源
4.1 模拟环境:
- Isaac Sim(NVIDIA的物理仿真环境;GPU加速;超快速):
- Sparrow(轻量级移动机器人模拟器;适合DRL):
- ROS(流行的综合性机器人物理仿真器;体积大且速度慢):
- Webots(流行的机器人物理仿真器;比ROS快;但真实性稍逊):
4.2 书籍:
- 《强化学习:导论》--理查德·S·萨顿
- 《深度学习入门:基于Python的理论与实现》--斋藤康毅
4.3 在线课程:
- RL课程(bilibili)--李宏毅
- RL课程(Youtube)--李宏毅
- UCL的RL课程--戴维·西尔弗
- 动手强化学习--上海交通大学
- DRL课程--王树森
4.4 博客:
- OpenAI Spinning Up
- 策略梯度定理 --苍溪
- 策略梯度算法 --莉莲
- PPO定理
- 近端策略优化的37个实现细节
- 优先级经验回放
- 软演员评论家
- 对强化学习的深入观察 --莉莲
- TD3简介
5. 重要论文
DQN:Mnih V, Kavukcuoglu K, Silver D, 等. 通过深度强化学习实现人类水平的控制[J]. 自然, 2015, 518(7540): 529-533.
Double DQN:Van Hasselt H, Guez A, Silver D. 深度强化学习中的双Q学习[C]//美国人工智能协会人工智能会议论文集. 2016, 30(1).
Duel DQN:Wang, Ziyu, 等. “用于深度强化学习的决斗网络架构”. 国际机器学习大会. PMLR, 2016.
PER:Schaul T, Quan J, Antonoglou I, 等. 优先级经验回放[J]. arXiv预印本 arXiv:1511.05952, 2015.
C51:Bellemare M G, Dabney W, Munos R. 强化学习的分布视角[C]//国际机器学习大会. PMLR, 2017: 449-458.
NoisyNet DQN:Fortunato M, Azar M G, Piot B, 等. 用于探索的噪声网络[J]. arXiv预印本 arXiv:1706.10295, 2017.
PPO:Schulman J, Wolski F, Dhariwal P, 等. 近端策略优化算法[J]. arXiv预印本 arXiv:1707.06347, 2017.
DDPG:Lillicrap T P, Hunt J J, Pritzel A, 等. 基于深度强化学习的连续控制[J]. arXiv预印本 arXiv:1509.02971, 2015.
TD3:Fujimoto S, Hoof H, Meger D. 解决演员-评论家方法中的函数逼近误差[C]//国际机器学习大会. PMLR, 2018: 1587-1596.
SAC:Haarnoja T, Zhou A, Abbeel P, 等. 软演员-评论家:具有随机演员的离策略最大熵深度强化学习[C]//国际机器学习大会. PMLR, 2018: 1861-1870.
ASL:通过部分解耦的强化学习和向量化多样性,在一小时内训练一个真实世界的局部路径规划器
ColorDynamic:适用于非结构化和动态环境的可泛化、可扩展、实时、端到端局部规划器
6. 引用
@misc{DRL-Pytorch,
author = {Jinghao Xin},
title = {DRL-Pytorch},
year = {2022},
publisher = {GitHub},
journal = {GitHub Repository},
howpublished = {\url{https://github.com/XinJingHao/DRL-Pytorch}},
}
7. 我的代码训练曲线:
Q-learning:
Duel Double DQN:
| CartPole | LunarLander |
|---|---|
![]() |
![]() |
Atari游戏上的噪声Duel DDQN:
| Pong | Enduro |
|---|---|
![]() |
![]() |
优先级DQN/DDQN:
| CartPole | LunarLander |
|---|---|
分类DQN:
| CartPole | LunarLander |
|---|---|
NoisyNet DQN:
| CartPole | LunarLander |
|---|---|
![]() |
![]() |
PPO离散版:
PPO连续版:
DDPG:
| Pendulum | LunarLander连续版 |
|---|---|
TD3:
SAC连续版:
SAC离散版:
演员-分享者-学习者(ASL):
常见问题
相似工具推荐
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 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器





