Fix gym Atari ROMs

main
Aurélien Geron 2021-10-18 16:30:03 +13:00
parent f8cf0148d2
commit 0b8a519395
1 changed files with 3 additions and 2 deletions

View File

@ -58,8 +58,9 @@
"\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,atari_py] AutoROM\n", " %pip install -U tf-agents pyvirtualdisplay\n",
" AUTOROM_PREFIX=\"/bin\n", " %pip install -U gym>=0.21.0\n",
" %pip install -U gym[box2d,atari,accept-rom-license]\n",
"\n", "\n",
"# Common imports\n", "# Common imports\n",
"import numpy as np\n", "import numpy as np\n",