安装
安装基本 PettingZoo 库: pip install pettingzoo
这并不包括所有环境系列的依赖项(某些环境在某些系统上的安装可能存在问题)。
要安装一个系列的依赖项,请使用 pip install 'pettingzoo[atari]' 或使用 pip install 'pettingzoo[all]' 安装所有依赖项。
支持 Linux 和 macOS 上的 Python 3.8、3.9、3.10 和 3.11。接受与 Windows 有关的 PR,但不正式支持它(经笔者验证Windows中python3.8时安装该环境,环境不会出现问题)。
注意:某些 Linux 发行版可能需要手动安装 cmake、swig 或 zlib1g-dev(例如,sudo apt install cmake swig zlib1g-dev)
windows安装注意: 按照简单的 pip install pettingzoo 由于网速等原因可能无法安装成功,这时可以选择指定某一清华源进行安装:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pettingzoo
PettingZoo 包括以下几类环境:
- Atari:雅达利, 多人Atari 2600 游戏(合作、竞争和混合);
- Butterfly: 合作图形游戏,需要高度的协调性;
- Classic: 经典游戏,包括纸牌游戏、棋盘游戏等;
- MPE: 一套简单的非图形通信任务;
- SISL:3 个合作环境。
与环境交互
from pettingzoo.butterfly import cooperative_pong_v5
#env = cooperative_pong_v5.env()
env = cooperative_pong_v5.env(render_mode="human") #默认render_mode=None
env.reset(seed=42) #随机种子设定
for agent in env.agent_iter():
observation, reward, termination, truncation, info