很多benchmark算法都是在Atari游戏上做实验,所以有必要学习使用Atari游戏环境做强化学习。这里说一下安装Atari游戏模块的方法以及安装中遇到的问题
安装步骤
pip install gym[atari]
pip uninstall atari_py
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上测试,发现可以运行。
贴一段测试代码:
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