在Jupyter笔记本上运行/呈现OpenAI Gym的包装器
gym-notebook-wrapper的Python项目详细描述
健身笔记本包装
健身房笔记本包装提供了小包装运行和渲染 OpenAI Gym在Jupyter Notebook或类似(例如Google Colab)。在
1。要求
- Linux系统
- Xvfb
- 在Ubuntu上,你可以安装
sudo apt update && sudo apt install xvfb
。在
- 在Ubuntu上,你可以安装
- 开放式GL(针对某些环境)
- 在Ubuntu上,你可以安装
sudo apt update && sudo apt install python-opengl
- 在Ubuntu上,你可以安装
2。安装
您可以从安装
PyPI带pip install gym-notebook-wrapper
3。使用
本文中的gnwrapper
模块中实现了三个类
健身笔记本包装袋。在
3.1简单的一次性动画
用gnwrapper.Animation
包装gym.Env
类。这就是全部!这个
render()
方法在其输出上显示环境。示例代码
以下是:
3.1.1代码
importgnwrapperimportgymenv=gnwrapper.Animation(gym.make('CartPole-v1'))obs=env.reset()for_inrange(1000):next_obs,reward,done,info=env.step(env.action_space.sample())env.render()obs=next_obsifdone:obs=env.reset()
3.1.2限制
- 调用
render()
方法删除同一单元格的其他输出。在 - 输出图像只显示一次。在
3.2循环动画
用gnwrapper.LoopAnimation
包装gym.Env
类。这个包装纸
调用render()
方法时存储显示图像,并显示
通过display(dpi=72,interval=50)
方法循环动画。在
3.2.1代码
^{pr2}$3.2.2限制
- 需要大量内存来存储和显示大的显示步骤
- 可能引发内存错误
3.3电影动画
用gnwrapper.Monitor
包装gum.Env
类。此包装继承
gym.wrappers.Monitor
并实现用于嵌入的{
如果您调用display(reset=True)
,视频列表将被清除,并且
下一个display()
方法只显示新视频。在
3.3.1代码
importgnwrapperimportgymenv=gnwrapper.Monitor(gym.make('CartPole-v1'),directory="./")o=env.reset()for_inrange(100):o,r,d,i=env.step(env.action_space.sample())ifd:env.reset()env.display()
3.3.2限制
- 需要磁盘空间来保存电影
4。注释
gnwrapper.Animation
和{gym.Wrapper
,以便它可以访问
gym.Env
和{action_space
)。在
5。链接
- 项目
标签: