Python 版 Vertex AI SDK 简介

借助 Python 版 Vertex AI SDK,您可以在 Vertex AI 中自动注入数据、训练模型,并获取预测结果。Vertex AI SDK 通过 Python 代码来访问 Vertex AI API,让您能以编程方式完成可在 Google Cloud 控制台中执行的大部分操作。

如需了解如何安装 Python 版 Vertex AI SDK,请参阅安装 Python 版 Vertex AI SDK。如需查看 Vertex AI SDK 参考指南,请参阅 Vertex AI SDK 参考文档

为何使用 Vertex AI SDK

如果您是经验丰富的机器学习 (ML) 及人工智能 (AI) 工程师或是希望以编程方式自动执行工作流的数据科学家,则推荐您使用 Python 版 Vertex AI SDK。Python 版 Vertex AI SDK 与 Vertex AI Python 客户端库类似,只不过 Vertex AI SDK 内容更具概括性,没有那么细化。如需了解详情,请参阅了解 SDK 和客户端库的区别

使用 Python 版 Vertex AI SDK 编写代码

如需使用 Python 版 Vertex AI SDK,请执行以下操作:

  1. 通过在虚拟环境中运行以下命令,安装 google-cloud-aiplatform 软件包,其中包含 Python 版 Vertex AI SDK 和 Vertex AI Python 客户端库:

    pip install --upgrade google-cloud-aiplatform
    
  2. 使用以下代码导入 google.cloud.aiplatform 命名空间:

    from google.cloud import aiplatform
    

了解 Python 版 Vertex AI SDK

请参阅以下文档:

  • Vertex AI SDK 类概览:介绍 Vertex AI SDK 中的关键类和功能。

  • Vertex AI Python 参考文档:包含 google-cloud-aiplatform 软件包中所有命名空间、类、方法和属性的参考文档,其中包含 Vertex AI SDK、Vertex AI SDK 预览版和 Vertex AI 客户端库。

试用代码示例和教程

Jupyter 笔记本教程介绍了如何在较大的工作流中使用 Python 版 Vertex AI SDK。如需了解详情,请参阅 Vertex AI 笔记本教程

Python 版 Vertex AI SDK GitHub 代码库中的代码示例展示如何完成各项任务。如需了解详情,请参阅 Python 版 Vertex AI SDK GitHub 代码库

了解 Vertex AI SDK 与其客户端库之间的差异

安装 Python 版 Vertex AI SDK 时,会同时安装 Vertex AI Python 客户端库。Vertex AI SDK 和 Vertex AI Python 客户端库具有粒度级别不同的类似功能。Vertex AI SDK 以比客户端库更高的抽象层运行,适合大多数常见的数据科学工作流。如果您需要较低级别的功能,请使用 Vertex AI Python 客户端库。

Vertex AI SDK 适用于 Python,Vertex AI 客户端库适用于 Python、Java 和 Node.js。如需了解如何安装 Java 或 Node.js 客户端库,请参阅安装 Vertex AI 客户端库。如果您的首选编程语言不支持客户端库,则可以使用 Vertex AI REST API。如需了解详情,请参阅 Vertex AI REST 参考文档

结合使用 Vertex AI Python 客户端库和 SDK

如果您在使用 Python 版 Vertex AI SDK 时发现自己需要更高的灵活性或掌控权,抑或需要使用未包含在 Vertex AI SDK 中的方法,您便可以在同一工作流中使用 Vertex AI Python 客户端库来解决这些问题。Vertex AI Python 客户端库使用不同的命名空间来访问 Vertex AI API。您可以在同一 Python 脚本中使用客户端库命名空间和 Python 版 Vertex AI SDK 命名空间,只需在 Python 脚本中为这两者各添加一行 import 内容即可。

导入 Vertex AI Python 客户端库命名空间

Vertex AI Python 客户端库命名空间为 google.cloud.aiplatform.gapic。此命名空间映射到 google.cloud.aiplatform_v1 命名空间。这两个命名空间可以互换使用。如需导入 Python 客户端库,请在 Python 脚本中添加以下任一项:

from google.cloud import aiplatform_v1
from google.cloud.aiplatform import gapic

后续步骤