azure.md 1.5 KB
Newer Older
S
Shaden Smith 已提交
1 2
---
title: "Getting Started with DeepSpeed on Azure"
3
tags: getting-started
S
Shaden Smith 已提交
4 5
---

6
This tutorial will help you get started with DeepSpeed on Azure.
S
Shaden Smith 已提交
7 8 9

If you don't already have an Azure account please see more details here: [https://azure.microsoft.com/](https://azure.microsoft.com/).

10
# DeepSpeed on Azure via AzureML
11

12
The recommended and simplest method to try DeepSpeed on Azure is through [AzureML](https://azure.microsoft.com/en-us/services/machine-learning/). Please take a look at easy-to-use examples for Megatron-DeepSpeed, Transformers and CIFAR training [here](https://github.com/Azure/azureml-examples/tree/main/python-sdk/workflows/train/deepspeed).
S
Shaden Smith 已提交
13

14
> Our [Megatron-DeepSpeed](https://github.com/microsoft/megatron-deepspeed) contains the most up to date [recipe](https://github.com/microsoft/Megatron-DeepSpeed/tree/main/examples/azureml) for end-to-end training on AzureML.
S
Shaden Smith 已提交
15

16
# DeepSpeed on Azure VMs
S
Shaden Smith 已提交
17

18
If you don't have access to AzureML or if want to build a custom environments using [Azure virtual machines](https://azure.microsoft.com/en-us/services/virtual-machines/) or Azure VM Scale-Sets ([VMSS](https://docs.microsoft.com/en-us/azure/virtual-machine-scale-sets/overview)), we are working on easy-to-use cluster setup scripts that will be published in the next few weeks.
S
Shaden Smith 已提交
19

20
If you already have a cluster setup, you can use the [azure recipes](https://github.com/microsoft/Megatron-DeepSpeed/tree/main/examples/azure) that can easily be modified to train various model configurations.