Python

Google Colab上で実行されているかどうかPythonで判定する

Posted on 2022年3月20日 (Last modified on 2023年1月14日) 1 min read  • 27 words
Google Colab上で実行されているかどうかPythonで判定する

この記事は何

Google Colaboratory上で動かすことが前提のコードを書く場合、Colaboratory上で提供されていないライブラリはユーザーが自分でインストールする必要があります。たとえば以下のページのコードではdtreevizというライブラリを使用していて、Colab上で実行するためにはインストールが必要です。

決定木のパラメータ - K_DM

そのため、Colab上で実行されているか判定してColab上の場合のみインストールを実行できないか調べました。

解決方法

Module: core.getipython — IPython 8.8.0 documentation

get_ipython()を使うとipythonの実行環境を取得できます。 これを使うことでcolab上で実行されているかを判定できました。

コード例:

if str(get_ipython()).startswith("<google.colab."):
    !pip install dtreeviz

colabで実行されている場合のみ !pip install dtreeviz が実行されてdtreevizがインストールされます。

¯\_(ツ)_/¯

こんちゃす。