强化学习—-雅达利游戏(atari)模块的安装

  • Post author:
  • Post category:其他


看了网上的部分教程,分别采用过这两个命令安装过,

pip install gym[atari]
pip install atari_py

然而在ide中使用时仍会报错。

后来发现,先执行:

pip uninstall atari_py

再执行:

pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py

就可以安装成功。

最后附上测试代码

import gym
from skimage import io
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
    io.imsave(jpgname,status)
    print(reward)
    if done: 
        print('dead in %d steps' % step)
        break
env.close()



版权声明:本文为qq_35027690原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。