最近我开始学习研究强化学习相关的知识,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跑起来了。
最后是一个实例程序:

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()