基于 Gymnasium 的自动驾驶仿真环境合集,用于训练和测试自动驾驶与决策智能体。
Farama-Foundation/PettingZoo
一个提供统一 API 和参考环境的 Python 多智能体强化学习研究库,接口风格类似 Gymnasium。
- Star
- 3,553
- Fork
- 555
- 未关闭 Issue
- 70
- 最近提交
- 8天前
这是什么?
PettingZoo 是一个用于多智能体强化学习研究的 Python 库,可以看作多智能体场景下的 Gymnasium。所有环境共享同一套接口,研究者只需编写一次算法,就能在多个任务上运行。
环境基于 Agent Environment Cycle(AEC)博弈模型设计,可以在同一 API 下支持回合制和同时行动的多种场景。对于智能体需要同时行动的游戏,项目还提供了 Parallel API。内置环境包括多人 Atari 2600 游戏、合作型图形游戏、经典纸牌和棋类游戏,以及一组合作型 SISL 任务。
基础包可通过 pip 安装,也可以按环境类别安装对应依赖,例如 pettingzoo[atari]。项目官方支持 Linux 和 macOS。配套库 SuperSuit 为 PettingZoo 和 Gymnasium 环境提供常用的强化学习封装器,文档中还包含与 CleanRL、Tianshou、AgileRL 配合训练的教程。
核心功能
- 01基于 AEC 博弈模型的统一多智能体环境 API
- 02适用于智能体同时行动场景的 Parallel API
- 03涵盖 Atari、合作图形游戏、经典棋牌和 SISL 任务的环境集合
- 04与 Gymnasium 相似的 reset、step 和迭代接口
- 05可按环境类别通过 pip extras 按需安装依赖
适用场景
- 在统一的环境集合上评测多智能体强化学习算法
- 快速原型验证合作型或竞争型智能体训练
- 基于标准接口开发自定义多智能体环境
- 使用 CleanRL、Tianshou 或 AgileRL 在 PettingZoo 任务上训练智能体
快速上手
- 1
安装 PettingZoo 基础库
pip install pettingzoo - 2
安装某一环境系列的依赖(例如 Atari)
pip install 'pettingzoo[atari]' - 3
创建环境并运行随机动作循环
使用前须知
- 仅官方支持 Linux 和 macOS,不officially 支持 Windows。
- 部分环境系列在某些系统上较难安装。
收录于专题
标签
常见问题
- PettingZoo 是什么?
- 一个提供统一 API 和参考环境的 Python 多智能体强化学习研究库,接口风格类似 Gymnasium。 PettingZoo 是一个用于多智能体强化学习研究的 Python 库,可以看作多智能体场景下的 Gymnasium。所有环境共享同一套接口,研究者只需编写一次算法,就能在多个任务上运行。
- PettingZoo 可以免费使用吗?用的什么许可证?
- PettingZoo 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
- PettingZoo 还在维护吗?
- 活跃维护中。PettingZoo 最近一次提交是在 2026年10月30日。
- PettingZoo 适合用来做什么?
- 在统一的环境集合上评测多智能体强化学习算法 快速原型验证合作型或竞争型智能体训练 基于标准接口开发自定义多智能体环境 使用 CleanRL、Tianshou 或 AgileRL 在 PettingZoo 任务上训练智能体
- PettingZoo 怎么安装和运行?
- 安装 PettingZoo 基础库: pip install pettingzoo 安装某一环境系列的依赖(例如 Atari): pip install 'pettingzoo[atari]' 创建环境并运行随机动作循环
- 在哪里获取 PettingZoo?
- 源代码和安装说明在 GitHub 上:https://github.com/Farama-Foundation/PettingZoo
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
面向强化学习与自动驾驶研究的轻量级 Python 驾驶仿真器,支持传感器仿真与 Gymnasium 风格环境。
基于 PyBullet 的 Gymnasium 环境,用于训练单机与多机四旋翼无人机的强化学习控制策略。
基于 Gymnasium API 与 MuJoCo 物理引擎的机器人强化学习仿真环境集合。
用于研究自动化攻防智能体在模拟企业网络中相互作用的实验平台,提供 OpenAI Gym 接口。
基于 Basilisk 仿真框架的 Python 包,用于构建航天器任务规划与调度的 Gymnasium 强化学习环境。