DeepMindLabのPython Module導入 - kerneltyu’s tech blog

前回の投稿から続いて今度はDeepMindLabのPython Moduleの導入をします。これによってOpenAIgymのような形で、pythonによるプログラムで環境から観測を受け取ったり、行動を環境に適用させたりすることができます。

Python Moduleの導入に関しては、下記に記述があるので、それにそって書きます。

インストール

$ git clone https://github.com/deepmind/lab.git && cd lab
b$ azel build -c opt --python_version=PY3 //python/pip_package:build_pip_package
./bazel-bin/python/pip_package/build_pip_package /tmp/dmlab_pkg
$ pip install /tmp/dmlab_pkg/deepmind_lab-1.0-py2-none-any.whl --force-reinstall

これでインストールができたかと思います。下記のテストコードを走らせると

import deepmind_lab
import numpy as np

# Create a new environment object.
lab = deepmind_lab.Lab("tests/empty_room_test", ['RGB_INTERLEAVED'],
                       {'fps': '30', 'width': '80', 'height': '60'})
lab.reset(seed=1)

# Execute 100 walk-forward steps and sum the returned rewards from each step.
print(sum(
    [lab.step(np.array([0,0,0,1,0,0,0], dtype=np.intc)) for i in range(0, 100)]))

おそらく、

ModuleNotFoundError: No module named 'dm_env'

というエラーに遭遇します。これは単純にdm_envというモジュールが無いだけなので落ち着いてpip installします。

pip install dm_env

これでテストコードが走るようになるかと思います。

以上！