测试笔记本是否在Google Colab上运行

Question

如何测试我的笔记本是否在Google Colab上运行？

我需要此测试，因为在便携式计算机或Colab上运行时，获取/解压缩训练数据的方式有所不同。

Answer 1

尝试导入google.colab

try:
  import google.colab
  IN_COLAB = True
except:
  IN_COLAB = False

或者只是检查它是否在sys.modules中

import sys
IN_COLAB = 'google.colab' in sys.modules

Answer 2

还可以检查使用的ipython解释器。我认为这更加清楚了，您不必导入任何模块。

if 'google.colab' in str(get_ipython()):
  print('Running on CoLab')
else:
  print('Not running on CoLab')

如果您需要多次执行此操作，则可能要分配一个变量，这样就不必重复str(get_ipython())。

RunningInCOLAB = 'google.colab' in str(get_ipython())

如果在Google Colab笔记本中运行，

RunningInCOLAB为True。

Answer 3

您可以像这样检查环境变量：

import os
if 'COLAB_GPU' in os.environ:
   print("I'm running on Colab")

实际上，您可以打印出os.environ来检查与colab相关的内容，然后检查密钥

Answer 4

在%%bash单元格中，使用：

%%bash
[[ ! -e /colabtools ]] && exit  # Continue only if running on Google Colab

# Do Colab-only stuff here

或者与Python等效

import os
if os.path.exists('/colabtools'):
  # do stuff