使用 Cloud Profiler 在使用预构建容器的自定义训练中分析模型训练性能:笔记本

在本教程中,您将学习如何在 Vertex AI 中针对使用预构建容器的自定义训练作业启用 Profiler。

笔记本:在预构建容器中使用 Cloud Profiler 分析模型训练性能

本教程使用以下 Google Cloud 机器学习服务和资源:

  • Vertex AI Training
  • Vertex AI TensorBoard

执行的步骤包括:

  • 准备自定义训练代码,并将训练代码作为 Python 软件包加载到预构建容器中。
  • 创建并运行启用了 Profiler 的自定义训练作业。
  • 查看 Profiler 信息中心以调试模型训练性能。

相关内容