最近我开始学习研究强化学习相关的知识，OpenAI开发的gym是一个很好的训练与测试的库，里面包含了很多Atari游戏，可以用于训练与测试Agent。但是犹豫gym的库进行了更新，想要让这些程序跑起来，需要做一点点准备。
首先是需要额外的库，需要通过pip安装ale_py，autorom这两个库，并运行AutoRom命令来下载Atari游戏。
如果还是提示没有游戏或者是环境，可以使用python3 -m atari_py.import_roms ROMS_DIR或者是ale-import-roms ROMS_DIR来导入游戏。游戏可以从网上找到下载地址。
通过这两个库，就可以让gym的Atari跑起来了。
最后是一个实例程序：

  • 01
  • 02
  • 03
  • 04
  • 05
  • 06
  • 07
  • 08
  • 09
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
import gym
import ale_py
import time

env = gym.make('BreakoutNoFrameskip-v4', render_mode='human')
print("Observation Space: ", env.observation_space)
print("Action Space       ", env.action_space)

env.metadata['render_fps'] = 30

obs = env.reset()
for i in range(1000):
    env.render()
    action = env.action_space.sample()
    obs, reward, terminated, truncated, info = env.step(action)
    time.sleep(0.01)
env.close()