web-dev-qa-db-fra.com

Comment puis-je exécuter correctement OpenAI gym avec nvidia-docker et voir les environnements

J'essaie donc de lancer OpenAI gym dans un conteneur de menu fixe, mais cela ressemble à ceci: enter image description here

Remarquez que la fenêtre de pong a un problème de rendu étrange où elle répète des choses et les couleurs sont éteintes. (En fait, si vous regardez le tableau renvoyé par env.reset (), vous pouvez voir les valeurs en pixels. Le problème réside donc dans le rendu, pas dans la transmission x. Voici les envahisseurs d'espace:

enter image description here

Ma configuration est très simple. - Je suis sur une installation locale d'ubuntu 16.04 avec un Nvidia gtx1060 et corei7 - J'ai installé le pilote de fichier de traitement nvida avec --no-opengl-files (selon les instructions de Nvidia et beaucoup d'autres). - Plus précisément, je suis en cours d'exécution image floydhub/pytorch docker.

Est-ce que quelqu'un reconnaît le problème de rendu particulier et ce que cela pourrait signifier? Cela ressemble presque à un StackOverflow d'un tampon d'image! Que puis-je faire pour localiser le bogue?

Vous pouvez facilement le reproduire comme suit:

docker run -it --user=$(id -u) --env="DISPLAY" --workdir="/home/$USER" --volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" floydhub/pytorch:0.1.11-gpu-py3.6 bash

Maintenant, dans l'image, tapez python, puis ce qui suit:

import gym
gym.make('Pong-v0').render()

Cela devrait ouvrir une fenêtre x-forwarded sur votre machine, mais l'affichage est corrompu (du moins pour moi)

image

Ci-dessus, j'ai effectivement utilisé SpaceInvaders-v0

5
AwokeKnowing

Ce problème fonctionnait sous une ancienne version de gym et une version plus récente de gym [atari]. Voici à quoi ressemblait mon rendu de pong ()

enter image description here

J'ai pu résoudre ce problème en m'assurant d'avoir installé la dernière version des deux en exécutant

pip install gym --upgrade
pip install gym[atari] --upgrade

Puis il s’affiche correctement

enter image description here

2
alexbhandari