nvidia-dockerでatariとかを動かす

Dockerfile

# $ docker build -t rlgpuenv .
# $ xhost +
# $ nvidia-docker run --rm \
#  -v $PWD:/dir \
#  -v /tmp/.X11-unix/:/tmp/.X11-unix/ \
#  -e DISPLAY=$DISPLAY \
#  -e QT_X11_NO_MITSHM=1 \
#  rlgpuenv python3 game.py

FROM nvidia/cuda
MAINTAINER ytbilly3636

RUN apt update
RUN apt install -y  python3-dev \
                    python3-pip \
                    python3-tk \
                    python-opengl \
                    libgtk2.0-dev
RUN pip3 install    chainer \
                    cupy \
                    chainerrl \
                    gym[atari] \
                    gym[box2d] \
                    matplotlib \
                    opencv-python

RUN mkdir /dir
WORKDIR /dir
VOLUME ["/dir"]

CMD ["bash"]