gym.render() でpygame is not installed ~ のエラーが出たときの対処法

困ったこと

Open AI Gym のenv.render() でエラーが発生してrender が使用できなくなった！というときに試す設定を書き残しておきます。

pygame is not installed ~ というエラーになる

使用したコードは下記の通りです。

import gym

env = gym.make('CartPole-v0')
observation = env.reset()
for i in range(100):
  env.render()
  observation, reward, done, info = env.step(1)
env.env.close()

下記のようにpygame is not installed ~ のようなエラーになる場合、

  File "/opt/anaconda3/envs/xxxx.py", line 199, in render
    raise DependencyNotInstalled(
gym.error.DependencyNotInstalled: pygame is not installed, run `pip install gym[classic_control]`

ターミナルで下記を追加してみましょう。

pip3 install pygame

参考

• https://bobbyhadz.com/blog/python-no-module-named-pygame