如何告诉 Azure Batch API 为池中的 VM 保留 VM 映像？

Question

我正在关注本教程https://docs.microsoft.com/en-us/azure/batch/tutorial-parallel-python，了解如何使用 Azure Batch API。

从那篇文章中不清楚，但是批处理运行完成后池中的节点会发生什么。

我关心的原因是我将使用的节点需要在批处理开始之前进行大量设置。是否可以让虚拟机在运行之间保留该设置以节省设置所需的带宽和时间？

此外，如果该设置需要重新启动（例如在安装 GPU 驱动程序之后），是否可以在使用集群之前执行此操作？

Answer 1

batch_client.pool.delete(_POOL_ID) 行会导致池被删除，这会导致其所有节点也被删除。 在运行之间保留 VM 的方法是不删除池，只需将下一个作业提交到同一个池。

关于您的广泛设置，包括 GPU 驱动程序和重新启动，我假设您是通过池启动任务执行此操作的。我认为在启动任务中重新启动（假设它是最后一个命令）应该没问题，尽管我没有尝试过。重启后该节点应该准备就绪。

也许更好的选择是use a custom VM image 已经包含所有复杂的设置，或者use a Docker container 也可以包括这样的复杂设置（包括安装 CUDA 等）。