gym.error.DependencyNotInstalled: No module named 'atari_py'. (HINT: you can install Atari dependenc

我的环境是ubuntu 16.04, python3

在运行一个policy gradient程序的时候，出现了如下的错误：

/home/eric/anaconda3/lib/python3.6/site-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
Using TensorFlow backend.
Traceback (most recent call last):
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/atari/atari_env.py", line 9, in 
    import atari_py
ModuleNotFoundError: No module named 'atari_py'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "pg.py", line 86, in 
    env = gym.make("Pong-v0")
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/registration.py", line 167, in make
    return registry.make(id)
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/registration.py", line 119, in make
    env = spec.make()
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/registration.py", line 85, in make
    cls = load(self._entry_point)
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/registration.py", line 14, in load
    result = entry_point.load(False)
  File "/home/eric/anaconda3/lib/python3.6/site-packages/pkg_resources/__init__.py", line 2408, in load
    return self.resolve()
  File "/home/eric/anaconda3/lib/python3.6/site-packages/pkg_resources/__init__.py", line 2414, in resolve
    module = __import__(self.module_name, fromlist=['__name__'], level=0)
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/atari/__init__.py", line 1, in 
    from gym.envs.atari.atari_env import AtariEnv
  File "/home/eric/anaconda3/lib/python3.6/site-packages/gym/envs/atari/atari_env.py", line 11, in 
    raise error.DependencyNotInstalled("{}. (HINT: you can install Atari dependencies by running 'pip install gym[atari]'.)".format(e))
gym.error.DependencyNotInstalled: No module named 'atari_py'. (HINT: you can install Atari dependencies by running 'pip install gym[atari]'.)

发现百度出来的方案都不work，google了一下，然后运行：

pip install atari-py

就可以了。

参考文献

[1].atari_py. https://github.com/openai/atari-py