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

尝试导入google.colab

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

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

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

还可以检查使用的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())

RunningInCOLAB 如果在Google Colab笔记本中运行,则为真。

您可以这样检查环境变量:

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

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

在一个 %%bash 单元格,使用:

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

# Do Colab-only stuff here

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