人工智能 (AI)、机器学习 (ML) 和高性能计算 (HPC) 工作负载通常需要您创建自定义映像,其中包含其他软件包并经过调整以实现最佳性能。您现在可以使用高级计算映像为 AI/ML 或 HPC 工作负载设置优化环境。
高级计算映像为 AI/机器学习和 HPC 基础设施提供标准化映像堆栈。这些映像可让您无需为 AI/机器学习和 HPC 工作负载手动构建自己的自定义映像。
优势
Advanced Compute Images 具有以下优势:
- 支持开箱即用的 AI/机器学习或 HPC 工作负载的计算实例。无需手动调整性能、管理实例重启、安装 Slurm 代理,也不必及时了解最新 Google Cloud 更新即可实现紧密耦合的 AI/ML 或 HPC 工作负载。
- 一致、可重现的性能。映像标准化可提供一致、可重现的应用级性能。
Advanced Compute Images 的类型
高级计算映像包含多个配置层,可支持运行 AI/机器学习和 HPC 工作负载。
- 操作系统配置:包括停用自动更新、设置经过 HPC 优化的 ulimit、调整内核 sysctl 参数,以及配置 SELinux 等安全设置。
- 网络调优:包含网络调优所需的脚本,例如启用多队列服务。
- Google Cloud 集成:安装并配置 Google Cloud CLI 和 Ops 代理。
- 容器工具:安装并配置所有与容器相关的软件,包括:
- Docker
- NVIDIA Container Toolkit(版本 1.19.0-1)
- NVIDIA Enroot
- NVIDIA Pyxis
- NVIDIA:安装以下工具:
- MPI:安装消息传递接口 (MPI) 库。 在基于 Ubuntu 的映像上安装 Open MPI。
- Slurm:安装创建 Slurm 集群节点所需的核心组件,包括:
- Slurm 二进制文件(版本 25.11)
- 用于身份验证的 Munge
- 用于进程管理的 PMIx
- JSON Web 令牌 (JWT) 库 (
libjwt)
- 文件系统客户端:安装用于访问各种文件系统的客户端工具,例如 Cloud Storage FUSE、NFS 实用程序 (
nfs-utils) 和 Lustre 客户端。 - 开发者工具:安装常见的开发和调试工具链以及实用程序,例如:
- 环境管理:安装和配置环境管理工具(主要是 Lmod),并设置必要的配置文件脚本,以便用户可以使用模块命令。
支持的硬件和映像系列
高级计算映像支持 GPU 工作负载,并具有以下特征:
- 针对 NVIDIA 加速器进行了优化
- 支持 Ubuntu LTS 22.04 和 Ubuntu LTS 24.04
- 预集成 CUDA 和 NVIDIA 驱动程序
- 适合 AI、机器学习或 HPC 工作负载
| 硬件 | 操作系统 | 编排程序 | 特性 | 架构 | 支持的机器系列 | 映像系列 |
|---|---|---|---|---|---|---|
| GPU | Ubuntu LTS 22.04 | Slurm 25.11 |
|
AMD x86_64 | A3 Ultra、A4 | aci-gpu-u2204-slurm-2511-cuda-130-nvidia-580-amd64 |
| GPU | Ubuntu LTS 24.04 | Slurm 25.11 |
|
AMD x86_64 | A4 | aci-gpu-u2404-slurm-2511-cuda-130-nvidia-580-amd64 |
| GPU | Ubuntu LTS 24.04 | Slurm 25.11 |
|
Arm64 | A4X | aci-gpu-u2404-slurm-2511-cuda-130-nvidia-580-arm64 |
| GPU | Ubuntu LTS 22.04 | 无 |
|
AMD x86_64 | A3 Ultra、A4 | aci-gpu-u2204-cuda-130-nvidia-580-amd64 |
| GPU | Ubuntu LTS 24.04 | 无 |
|
AMD x86_64 | A3 Ultra、A4 | aci-gpu-u2404-cuda-130-nvidia-580-amd64 |
预安装的软件包(或 RPM)
高级计算映像预安装了各种工具、库、驱动程序和 RPM。如需查看任何映像的已安装内容列表,请参阅相应映像的清单文件。
您可以在 /usr/share/google/manifest.txt 找到清单文件。
价格
高级计算映像无需额外付费即可使用。由于高级计算映像在 Compute Engine 上运行,因此 Compute Engine 资源(例如 vCPU、GPU、TPU、磁盘和内存)可能会产生费用。如需了解详情,请参阅 Compute Engine 价格。