google-colaboratory - 测试 notebook 是否在 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