rl-baselines3-zoo

环境要求

• Python 环境（可运行 pip）
• 源码安装需先获取仓库代码
• 完整安装需系统依赖 swig、cmake、ffmpeg
• macOS 安装 pybullet 时可能需要额外编译参数

安装与启动步骤

1. 1源码最小安装

   在项目根目录执行可编辑安装，只装核心依赖。适合只做基础训练、评估与调参的用户。

   pip install -e .

   复制

2. 2作为包安装

   不想拉源码时，直接把 rl_zoo3 作为 Python 包安装，随后即可使用其命令行接口。

   pip install rl_zoo3

   复制

3. 3安装系统依赖

   完整安装前先装编译与多媒体工具，swig/cmake 用于构建扩展环境，ffmpeg 用于录制视频。

   apt-get install swig cmake ffmpeg

   复制

4. 4完整扩展安装

   带 plots、tests、extras 附加项安装，获得绘图、测试以及额外环境支持。

   pip install -e .[plots,tests,extras]

   复制

5. 5macOS 失败重试

   macOS 上 pybullet 因 stdio.h 相关报错编译失败时，加上该 CFLAGS 重新安装即可。

   CFLAGS="-fno-define-target-os-macros" pip install -e .[plots,tests,extras]

   复制

如何确认成功

在任意目录运行 python -m rl_zoo3.train 或 rl_zoo3 train，能正常拉起训练脚本即安装成功。

常见问题

注意事项

• 完整安装前必须先装好 swig、cmake、ffmpeg 系统依赖，否则扩展环境可能构建失败。
• 日常使用只需最小安装即可；绘图、测试和额外环境才需要 [plots,tests,extras] 附加项。
• 自定义配置、超参数搜索等进阶用法请查阅官方文档 rl-baselines3-zoo.readthedocs.io。
• 项目仍在征集贡献者来补全经过调优的超参数集合。