很多benchmark算法都是在Atari游戏上做实验,所以有必要学习使用Atari游戏环境做强化学习。这里说一下安装Atari游戏模块的方法以及安装中遇到的问题

安装步骤

  1. pip install gym[atari]
  2. pip uninstall atari_py
  3. pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py

第一步做完后也会安装atari-py-0.2.6,然后在运行

 import gym
 env = gym.make('SpaceInvaders-v0')

的时候报错了:
……ctypes_init_.py", line 348, in init
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] 找不到指定的模块。

然后做第二步和第三步,把atari-py-0.2.6 替换成atari-py-1.2.2,再在python上测试,发现可以运行。

SpaceInvaders
贴一段测试代码:

import gym
env = gym.make('SpaceInvaders-v0')
status = env.reset()

for step in range(1000):
    env.render()
    thisstep = 1
    status, reward, done, info = env.step(thisstep)
    jpgname = './pic-%d.jpg' % step
    print(reward)
    if done:
        print('dead in %d steps' % step)
        break
env.close()

如果运行后出现了上面飞机大战的界面,就说明安装成功了,接下来就用Atari来测试各种强化学习的代码吧!

参考资料:
1.https://blog.csdn.net/qq_35027690/article/details/103859400

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐