第 2 章 管理自定义培训镜像

Ray CUDA

如果您使用计算密集型模型，而您想要使用 NVIDIA GPU 支持加速培训作业，您可以使用 Ray Compute Unified Device Architecture (CUDA)基础镜像来获得对 NVIDIA CUDA 工具包的访问。使用这个工具包，您可以使用为 NVIDIA GPU 优化的库和工具来加快您的工作。

Ray ROCm

如果您使用计算密集型模型，而您想要使用 AMD GPU 支持加速培训作业，您可以使用 Ray ROCm 基础镜像来获得对 AMD ROCm 软件堆栈的访问。使用这个软件堆栈，您可以使用为 AMD GPU 优化的库和工具来加快您的工作。

KFTO CUDA

如果您使用计算密集型模型，而您想要使用 NVIDIA GPU 支持加快培训作业，您可以使用 Kubeflow training Operator (KFTO) CUDA 基础镜像来访问 NVIDIA CUDA 工具包。使用这个工具包，您可以使用为 NVIDIA GPU 优化的库和工具来加快您的工作。

KFTO ROCm

如果您使用计算密集型模型，而您想要使用 AMD GPU 支持加速培训作业，您可以使用 KFTO ROCm 基础镜像来获得对 AMD ROCm 软件堆栈的访问。使用这个软件堆栈，您可以使用为 AMD GPU 优化的库和工具来加快您的工作。