如何在顶点AI中为用户管理的笔记本设置环境变量

Question

我试图在google云顶点AI中为用户管理的笔记本电脑设置一些环境变量。我不想从jupyter笔记本本身设置这个值，因为我希望这些环境变量对任何从这个笔记本实例打开jupyter笔记本的人都可用。这是我迄今所尝试过的，但没有任何效果：

1. 我有一个现有的用户管理笔记本。我在笔记本vm中设置了一个环境变量，即export TEST_VAR=TEST_VARIABLE_WAS_SET。但是，当我从控制台打开一个jupyter笔记本并执行os.environ["TEST_VAR"]时，它会给出一个关键错误。因此，我假设这与以下事实有关:顶点AI启动的jupyter实验室会话处于不同的shell会话或类似的会话中。我还尝试将以下两个元数据键添加到vm中，然后重新启动vm，但是它没有工作：

gcloud compute instances add-metadata ${INSTANCE_NAME} --metadata startup-script-url=$GCS_BUCKET_NAME/script.sh

其中script.sh是：

#!/bin/bash

export TEST_VAR=TEST_VARIABLE_WAS_SET

和

gcloud compute instances add-metadata ${INSTANCE_NAME} --metadata container-env-file=$GCS_BUCKET_NAME/notebook-env.txt

哪里的笔记本-env.txt是

TEST_VAR=TEST_VARIABLE_WAS_SET

1. I还尝试从云控制台创建一个用户托管笔记本的新实例。在这个过程中，我试图在“script，以便在创建后运行”中提供一个脚本，并通过提供“元数据”选项，提供作为startup-script-url的键和在Google云存储中作为脚本位置的值。该脚本是之前相同的启动脚本。

那么，对于现有的用户管理笔记本，以及当我创建新的笔记本时，如何实现这一点呢？

Answer 1

基于几天前我与您共享的公共文档，要正确设置环境变量并避免出现问题，您需要做的是在笔记本实例完全启动后运行脚本(使用--post-startup-script)。该路径必须是一个URL或云存储路径，正如前面提到的(gs://path-to-file/file-name).

以下面的方法为例；

–-post-startup-script=$GCS_BUCKET_NAME/script.sh