在Azure上开始使用DeepSpeed

本教程将帮助您在Azure上开始使用DeepSpeed。

如果您还没有Azure账户,请在此处查看更多详情:https://azure.microsoft.com/

通过AzureML在Azure上使用DeepSpeed

在Azure上尝试DeepSpeed的推荐且最简单的方法是通过AzureML。使用AzureML v2的训练示例和DeepSpeed自动调优示例可以在这里找到。

有关AzureML v1的示例,请查看Megatron-DeepSpeed、Transformers和CIFAR训练的易用示例这里

我们的Megatron-DeepSpeed包含了在AzureML上进行端到端训练的最新配方

Azure虚拟机上的DeepSpeed

如果您无法访问AzureML,或者想要使用Azure虚拟机或Azure VM Scale-Sets(VMSS)构建自定义环境,我们正在开发易于使用的集群设置脚本,这些脚本将在接下来的几周内发布。

如果您已经有一个集群设置,您可以使用azure recipes,这些配方可以轻松修改以训练各种模型配置。

更新: