Repolode
Farama-Foundation

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. 1

    安装 PettingZoo 基础库

    pip install pettingzoo
  2. 2

    安装某一环境系列的依赖(例如 Atari)

    pip install 'pettingzoo[atari]'
  3. 3

    创建环境并运行随机动作循环

使用前须知

  • 仅官方支持 Linux 和 macOS,不officially 支持 Windows。
  • 部分环境系列在某些系统上较难安装。

收录于专题

标签

#reinforcement-learning#multi-agent#gymnasium#python#simulation#research#api#multi-agent-reinforcement-learning#multiagent-reinforcement-learning

常见问题

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 数据每日更新。

相关项目