yandexdataschool / Practical_RL

A course in reinforcement learning in the wild

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Colab no longer bundles Atari ROMs in their Gym installation

dniku opened this issue · comments

https://www.coursera.org/learn/practical-rl/discussions/all/threads/vKyrK8EnEeu8TRLTS18hVQ

tl;dr:

import gym
gym.make('BreakoutNoFrameskip-v4')

crashes with

Exception: ROM is missing for breakout, see https://github.com/openai/atari-py#roms for instructions

The thread also proposes a solution.

I've tested the proposed solution and it works. Here is a slightly cleaned up version:

!wget http://www.atarimania.com/roms/Roms.rar
!pip install unrar
!mkdir ./roms
!unrar x Roms.rar ./roms
!python -m atari_py.import_roms ./roms

We could put it in setup_colab.sh.