安装 Vertex AI SDK for Python

使用 Vertex AI SDK for Python 自动执行机器学习 (ML) 工作流。本主题介绍如何安装 Vertex AI SDK for Python。如需详细了解 Vertex AI SDK，请参阅以下资源：

• 如需了解 Vertex AI SDK for Python，请参阅 Vertex AI SDK for Python 简介。
• 如需了解如何使用 Vertex AI SDK for Python 训练模型，请参阅使用 Vertex AI 和 Python SDK 训练模型。
• 如需了解 Vertex AI SDK for Python 中的类和方法，请参阅 Vertex AI SDK 参考文档。

安装 Vertex AI SDK for Python 包括以下步骤：

1. 创建一个隔离的 Python 环境
2. 安装 Vertex AI SDK 软件包
3. 初始化 Vertex AI SDK

创建一个隔离的 Python 环境

Python 最佳做法是在每个项目的独立 Python 环境中安装 Vertex AI SDK。这有助于防止依赖项、版本和权限冲突。您可以创建隔离的环境，以在 shell 中使用命令行或使用笔记本。

如需在使用命令行时创建隔离的环境，请激活 venv 环境。激活 venv 环境后，您就可以安装 Vertex AI SDK 并运行 Python 脚本了。如需了解详情，请参阅使用 venv 隔离依赖项和设置 Python 开发环境。

如需在隔离的环境中使用笔记本，您可以创建 Vertex AI Workbench 实例。然后，在 Vertex AI Workbench 实例的笔记本中安装 Vertex AI SDK 并运行 Python 脚本。如需了解详情，请参阅创建 Vertex AI Workbench 实例。

安装或更新 Vertex AI SDK 软件包

如需安装或更新 Vertex AI SDK，请在虚拟环境中运行以下命令：

pip install --upgrade google-cloud-aiplatform

初始化 Vertex AI SDK

安装 Vertex AI SDK for Python 后，您必须使用 Vertex AI 和 Google Cloud 详细信息初始化 SDK。例如，初始化 SDK 时，您可以指定项目名称、区域和暂存 Cloud Storage 存储桶等信息。以下示例展示了初始化 Vertex AI SDK 的方法。

def init_sample(
    project: Optional[str] = None,
    location: Optional[str] = None,
    experiment: Optional[str] = None,
    staging_bucket: Optional[str] = None,
    credentials: Optional[google.auth.credentials.Credentials] = None,
    encryption_spec_key_name: Optional[str] = None,
    service_account: Optional[str] = None,
):

    import vertexai

    vertexai.init(
        project=project,
        location=location,
        experiment=experiment,
        staging_bucket=staging_bucket,
        credentials=credentials,
        encryption_spec_key_name=encryption_spec_key_name,
        service_account=service_account,
    )

后续步骤

• 详细了解 Vertex AI SDK。