如何获取Vertex AI Notebook当前登录的GCP用户账号？

要获取当前使用Vertex AI Notebook的GCP用户账号（格式如my_gcp_user@my_domain.com），可以通过以下两种方式实现：

方法一：读取环境变量 #

Vertex AI Notebook会自动设置JUPYTER_SERVER_USER环境变量，直接读取该变量即可得到登录的GCP用户账号：

import os
gcp_user = os.environ.get("JUPYTER_SERVER_USER")
print(f"GCP用户账号 = {gcp_user}")

方法二：调用Metadata服务器 #

通过GCP Metadata服务器的特定端点，获取创建Notebook实例的用户信息：

curl "http://metadata.google.internal/computeMetadata/v1/instance/attributes/created-by" -H "Metadata-Flavor: Google"

返回结果即为符合要求格式的GCP用户账号。

注意事项：

• 上述方法获取的是登录/创建Notebook的GCP用户账号，区别于服务账号或系统本地用户。
• Notebook实例默认具备访问Metadata服务器的权限，无需额外配置。

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