为 Ray on Vertex AI 进行设置

在开始使用 Ray on Vertex AI 之前，请按照以下步骤设置 Google 项目和 Vertex AI SDK for Python：

按照设置项目和开发环境中的步骤为项目设置结算，安装 gcloud CLI 并启用 Vertex AI API。

启用 Vertex AI API
前提条件：您应该了解如何使用开源 Ray 开发程序。
此处使用的 Python 版 Ray on Vertex AI SDK 是 Python 版 Vertex AI SDK 的一个版本，其中包含 Ray 客户端、Ray BigQuery 连接器、Vertex AI 上的 Ray 集群管理和 Vertex AI 上的预测。
- 如果您在 Google Cloud 控制台中使用 Ray on Vertex AI，则 Colab Enterprise 笔记本将在您创建 Ray 集群后指导您完成 Python 版 Vertex AI SDK 安装过程。
- 如果您在 Vertex AI Workbench 或其他交互式 Python 环境中使用 Ray on Vertex AI，请安装 Python 版 Vertex AI SDK：
```
# The latest image in the Ray cluster includes Ray 2.9
# The latest supported Python version is Python 3.10.
$ pip install google-cloud-aiplatform[ray]
```
  安装 SDK 后，请先重启内核，然后再导入软件包。
  
  注意：如果您将 Vertex AI Workbench 笔记本用作客户端环境并使用 Deep Learning VM 作为机器映像，则 TensorFlow 企业版已预装 Ray 和 Python 版 Vertex AI SDK
（可选）如果您计划从 BigQuery 读取数据，则需要创建新的 BigQuery 数据集或使用现有数据集。
注意：如果您在 Vertex AI 上的 Ray 集群中运行与 BigQuery 等 Google 服务交互的代码，则 Vertex AI Custom Code Service Agent 用于进行身份验证。
（可选）为了降低 Vertex AI 中发生数据渗漏的风险，您可以在创建集群时启用 VPC Service Controls 并指定 VPC 网络。如需了解详情，请参阅将 VPC Service Controls 与 Vertex AI 搭配使用。

如果启用 VPC Service Controls，您将无法访问边界外的资源，例如 Cloud Storage 存储桶中的文件。

注意：Ray on Vertex AI 的最佳设置是每个项目一个自动模式 VPC 网络。如果您使用自定义模式 VPC 网络或使用多个 VPC 网络在同一项目中创建集群，则可能会遇到问题。
（可选）如需使用自定义容器映像，请将其托管在 Artifact Registry 上。通过自定义映像，您可以添加未包含在预构建容器映像中的 Python 依赖项。如需构建自定义映像，请参阅 Docker 文档中的“打包软件”。
（可选）如果您在 Vertex AI 上创建 Ray 集群时指定 VPC 网络，建议您在项目中设置一个自动模式 VPC 网络。如果您使用自定义模式 VPC 网络或使用多个 VPC 网络在同一项目中创建集群，则可能会遇到问题。

保护您的集群

遵循 Ray 最佳做法和指南，包括在可信网络上运行可信代码，以保护您的 Ray 工作负载。您的云实例中的 ray.io 部署属于共担责任模型。

如需详细了解 Google Cloud 最佳做法，请参阅 GCP-2024-020 安全公告。

支持的位置

自定义模型训练的特征可用性表列出了 Ray on Vertex AI 的可用位置。

后续步骤

在 Vertex AI 上创建 Ray 集群