借助 Vertex AI SDK for Python,您可以在 Vertex AI 中自动注入数据、训练模型,并获取预测结果。Vertex AI SDK 通过 Python 代码来访问 Vertex AI API,让您能以编程方式完成可在 Google Cloud 控制台中执行的大部分操作。
如需了解如何安装或更新 Vertex AI SDK for Python,请参阅安装 Vertex AI SDK for Python。如需了解详情,请参阅 Vertex AI SDK for Python API 参考文档。
为何使用 Vertex AI SDK
如果您是经验丰富的机器学习 (ML) 及人工智能 (AI) 工程师或是希望以编程方式自动执行工作流的数据科学家,则推荐您使用 Vertex AI SDK for Python。Vertex AI SDK for Python 与 Vertex AI Python 客户端库类似,只不过 Vertex AI SDK 内容更具概括性,没有那么细化。如需了解详情,请参阅了解 SDK 和客户端库的区别。
使用 Vertex AI SDK for Python 编写代码
如需使用 Vertex AI SDK for Python,请执行以下操作:
在虚拟环境中运行以下命令,安装
google-cloud-aiplatform
软件包(其中包含 Vertex AI SDK for Python 和 Vertex AI Python 客户端库):pip install --upgrade google-cloud-aiplatform
使用以下代码导入
google.cloud.aiplatform
命名空间:from google.cloud import aiplatform
-
If you're using a local shell, then create local authentication credentials for your user account:
gcloud auth application-default login
You don't need to do this if you're using Cloud Shell.
了解 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 笔记本教程介绍了如何在较大的工作流中使用 Vertex AI SDK for Python。如需了解详情,请参阅 Vertex AI 笔记本教程。
Vertex AI SDK for Python GitHub 代码库中的代码示例展示如何完成各项任务。如需了解详情,请参阅 Python 版 Vertex AI SDK GitHub 代码库。
了解 Vertex AI SDK 与其客户端库之间的差异
安装 Vertex AI SDK for Python 时,会同时安装 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
如果您在使用 Vertex AI SDK for Python 时发现自己需要更高的灵活性或掌控权,抑或需要使用未包含在 Vertex AI SDK 中的方法,您便可以在同一工作流中使用 Vertex AI Python 客户端库来解决这些问题。Vertex AI Python 客户端库使用不同的命名空间来访问 Vertex AI API。您可以在同一 Python 脚本中使用客户端库命名空间和 Vertex AI SDK for Python 命名空间,只需在 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
后续步骤
了解如何选择训练方法。