如何检测Python解释器是否运行在Google Colaboratory环境？

我原本的代码运行在Raspberry Pi上，现在想在Google Colaboratory里分析Google Drive中的数据，不想下载数据到本地。需要根据运行环境修改文件路径，有没有办法用if语句直接识别Colab环境？比如if somecondition: PLATFORM=colab。

我目前的思路是用os.uname()通过排除本地PC和树莓派的方式判定Colab环境，代码如下：

# 检查运行环境是PC还是Raspberry Pi
if os.uname().nodename == 'raspberrypi':
    PLATFORM = 'Raspberry Pi'
elif 'sashi' in os.uname().nodename.lower():
    PLATFORM = 'DEV'
else:
    PLATFORM = 'COLAB'

if PLATFORM == 'Raspberry Pi':
    PROJECT_DIR = '/home/pi/Projects'
elif PLATFORM == 'DEV':
    PROJECT_DIR = '/Users/sashi/repos/'
elif PLATFORM == 'COLAB':
    PROJECT_DIR = '/content/drive/My Drive/Colab Notebooks/'

def some_function(relative_path):
    for f in os.path.join(PROJECT_DIR, relative_path):
        print(f)
        # 处理文件逻辑

Colaboratory环境下os.uname()的输出为：

posix.uname_result(sysname='Linux', nodename='46f38c0b1d62', release='5.15.107+', version='#1 SMP Sat Apr 29 09:15:28 UTC 2023', machine='x86_64')

直接识别Colab环境的可靠方法 #

你可以通过检查Colab特有的环境变量来直接判定，比排除法更稳定：

方法1：检查COLAB_RELEASE_TAG环境变量

Colab会自动设置COLAB_RELEASE_TAG这个专属环境变量，直接判断它是否存在即可：

import os

# 直接判定是否为Colab环境
if 'COLAB_RELEASE_TAG' in os.environ:
    PLATFORM = 'COLAB'
    PROJECT_DIR = '/content/drive/My Drive/Colab Notebooks/'
elif os.uname().nodename == 'raspberrypi':
    PLATFORM = 'Raspberry Pi'
    PROJECT_DIR = '/home/pi/Projects'
elif 'sashi' in os.uname().nodename.lower():
    PLATFORM = 'DEV'
    PROJECT_DIR = '/Users/sashi/repos/'

方法2：结合目录检查（可选）

如果需要确保Google Drive已挂载，可以同时检查/content/drive目录是否存在：

import os

def is_colab():
    return 'COLAB_RELEASE_TAG' in os.environ and os.path.exists('/content/drive')

if is_colab():
    # 若未挂载Drive，可添加挂载代码
    from google.colab import drive
    drive.mount('/content/drive')
    PLATFORM = 'COLAB'
    PROJECT_DIR = '/content/drive/My Drive/Colab Notebooks/'
# 其他环境判断逻辑...

为什么不推荐排除法？ #

Colab的nodename是随机生成的容器ID（比如示例中的46f38c0b1d62），虽然当前能通过排除树莓派和本地PC来判定，但不排除未来出现命名冲突的可能，直接检查Colab专属的环境变量是更稳妥的方案。

内容的提问来源于stack exchange，提问作者Sashi