如何在本地VS Code中使用Vertex AI托管笔记本并实现SSH连接

一、SSH连接Vertex AI托管笔记本步骤 #

Vertex AI托管笔记本没有直接暴露外部SSH端口，必须通过gcloud CLI建立加密隧道来连接，具体操作如下：

1. 确认gcloud CLI已完成授权并配置好目标项目，执行命令切换到对应区域：

   gcloud config set compute/region <你的区域ID>
   

2. 列出当前项目下的托管笔记本实例，确认目标实例状态为RUNNING：

   gcloud ai notebooks instances list
   

3. 启动SSH隧道并建立临时连接，替换<实例名称>为你的托管笔记本实例名：

   gcloud ai notebooks instances start-ssh --instance=<实例名称>
   

   执行后，命令行会输出本地转发的端口号（例如localhost:12345），同时自动生成SSH密钥（默认存储在~/.ssh/google_compute_engine）。
4. 在VS Code中打开Remote-SSH插件，点击左侧栏的「+」添加新SSH配置，粘贴以下内容（替换端口号和密钥路径）：

   Host vertex-managed-notebook
     HostName localhost
     Port <本地转发的端口号>
     User jupyter
     IdentityFile ~/.ssh/google_compute_engine
   

5. 选择刚创建的配置，VS Code会自动连接到托管笔记本，连接成功后即可访问远程文件系统、终端。

二、在VS Code中使用托管笔记本的Jupyter功能 #

连接成功后，可直接在本地VS Code中操作托管笔记本的Jupyter环境：

• 打开托管笔记本中的.ipynb文件，VS Code的Jupyter插件会自动识别远程内核
• 右下角的内核选择器中，会显示托管笔记本上的可用内核，选择后即可本地编辑并远程执行代码
• 如需调试，可直接使用VS Code的调试功能，调试会话会在托管笔记本的环境中运行

说明 #

你之前尝试的用户托管笔记本教程不适用，是因为两类笔记本的架构不同：用户托管笔记本允许配置外部IP和自定义SSH密钥，而托管笔记本是完全托管的服务，依赖gcloud CLI的专用命令来建立安全隧道，无法直接通过普通SSH方式连接。

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