Auto download atari ROMs

main
Aurélien Geron 2021-10-17 19:21:13 +13:00
parent dbaffa5e14
commit 98f107eaa0
3 changed files with 7 additions and 5 deletions

View File

@ -58,8 +58,8 @@
"\n", "\n",
"if IS_COLAB or IS_KAGGLE:\n", "if IS_COLAB or IS_KAGGLE:\n",
" !apt update && apt install -y libpq-dev libsdl2-dev swig xorg-dev xvfb\n", " !apt update && apt install -y libpq-dev libsdl2-dev swig xorg-dev xvfb\n",
" %pip install -q -U tf-agents pyvirtualdisplay gym[box2d]\n", " %pip install -q -U tf-agents pyvirtualdisplay gym[box2d,atari,accept-rom-license]\n",
" %pip install -q -U atari_py==0.2.5\n", " !AutoROM --quiet --accept-license\n",
"\n", "\n",
"# Common imports\n", "# Common imports\n",
"import numpy as np\n", "import numpy as np\n",
@ -3228,5 +3228,5 @@
} }
}, },
"nbformat": 4, "nbformat": 4,
"nbformat_minor": 2 "nbformat_minor": 4
} }

View File

@ -46,6 +46,8 @@ dependencies:
- tfx~=1.3.0 # platform to deploy production ML pipelines - tfx~=1.3.0 # platform to deploy production ML pipelines
- transformers~=4.11.3 # Natural Language Processing lib for TF or PyTorch - transformers~=4.11.3 # Natural Language Processing lib for TF or PyTorch
- urlextract~=1.4.0 # optionally used in chapter 3, exercise 4 - urlextract~=1.4.0 # optionally used in chapter 3, exercise 4
- ale-py~=0.7.2 # used only in chapter 17
- AutoROM~=0.4.2 # used only in chapter 17
# Specific lib versions to avoid conflicts # Specific lib versions to avoid conflicts
- attrs=20.3 - attrs=20.3

View File

@ -49,13 +49,13 @@ tensorflow-addons~=0.14.0
# There are a few dependencies you need to install first, check out: # There are a few dependencies you need to install first, check out:
# https://github.com/openai/gym#installing-everything # https://github.com/openai/gym#installing-everything
gym[Box2D]~=0.21.0 gym[Box2D,atari,accept-rom-license]~=0.21.0
atari-py~=0.2.9
# On Windows, install atari_py using: # On Windows, install atari_py using:
# pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py # pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py
tf-agents~=0.10.0 tf-agents~=0.10.0
ale-py~=0.7.2
##### Image manipulation ##### Image manipulation
Pillow~=8.4.0 Pillow~=8.4.0