Colab での pip install 後の再起動

Question

Stable-Baselines
というライブラリーをGoogle Colabolatory で使用する際に必ず、再起動しないといけません。setup.pyの設定と関係があるのかもしれませんが、良い方法を探しています。

!pip install stable-baselines==2.10
# Select Version
%tensorflow_version 1.x
from stable_baselines.common.cmd_util import make_atari_env

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-7-295064432f15> in <module>()
----> 1 from stable_baselines.common.cmd_util import make_atari_env
      2 
      3 

/tensorflow-1.15.2/python3.7/stable_baselines/common/cmd_util.py in <module>()
      7 from mpi4py import MPI
      8 import gym
----> 9 from gym.wrappers import FlattenDictWrapper
     10 
     11 from stable_baselines import logger

ImportError: cannot import name 'FlattenDictWrapper' from 'gym.wrappers' (/usr/local/lib/python3.7/dist-packages/gym/wrappers/__init__.py)

---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can

ここで、一旦
!pip uninstall stable-baselines==2.10.0
でアンインストールして、再び上記のコードを実行すると、エラーは出ますがランタイムを再起動後にエラーは消えます。
ちなみに、以下のモジュールは問題ないです。よろしくお願いいたします。

from stable_baselines.common.vec_env import DummyVecEnv

Answer 1

あまりよく原理はわかっていませんが、ここに書かれているTry it in a notebookのリンク先にあるおまじない

%tensorflow_version 1.x
!apt install swig cmake libopenmpi-dev zlib1g-dev
!pip install stable-baselines[mpi]==2.10.0 box2d box2d-kengz

を入れると、importがWARNINGはでるものの無事成功します。

Comments

[teefpc]

す、すごいです。本当に動きました。