You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
* Integration with Azure Machine Learning: Track your PyTorch experiments on Azure Machine Learning studio or using the SDK.
31
-
* The image is also available as a [Data Science Virtual Machine (DSVM)](https://azure.microsoft.com/products/virtual-machines/data-science-virtual-machines/). To learn more about Data Science Virtual Machines, see [the DSVM overview documentation](data-science-virtual-machine/overview.md).
32
-
* Azure customer support reduces training and deployment latency.
33
-
* Improves training and deployment success rate.
34
-
* Avoid unnecessary image builds.
35
-
* Only have required dependencies and access right in the image/container.
25
+
***Flexibility**: Use as-is with preinstalled packages or build on top of the curated environment.
26
+
***Ease of use**: All components are installed and validated against dozens of Microsoft workloads to reduce setup costs and accelerate time to value.
27
+
***Efficiency**: Avoid unnecessary image builds and only have required dependencies that are accessible right in the image/container.
28
+
***Optimized training framework**: Set up, develop, and accelerate PyTorch models on large workloads, and improve training and deployment success rate.
29
+
***Up-to-date stack**: Access the latest compatible versions of Ubuntu, Python, PyTorch, CUDA/RocM, etc.
30
+
***Latest training optimization technologies**: Make use of [ONNX Runtime](https://onnxruntime.ai/) , [DeepSpeed](https://www.deepspeed.ai/), [MSCCL](https://github.com/microsoft/msccl), and more.
31
+
***Integration with Azure Machine Learning**: Track your PyTorch experiments on Azure Machine Learning studio or using the SDK. Azure customer support also reduces training and deployment latency.
32
+
***Availability as DSVM**: The image is also available as a [Data Science Virtual Machine (DSVM)](https://azure.microsoft.com/products/virtual-machines/data-science-virtual-machines/). To learn more about Data Science Virtual Machines, see [the DSVM overview documentation](data-science-virtual-machine/overview.md).
36
33
37
34
>[!IMPORTANT]
38
35
> To view more information about curated environment packages and versions, visit the Environments tab in the Azure Machine Learning [studio](./how-to-manage-environments-in-studio.md).
39
36
40
37
## Supported configurations for Azure Container for PyTorch (ACPT)
41
38
42
-
**Description**: The Azure Curated Environment for PyTorch is our latest PyTorch curated environment. It's optimized for large, distributed deep learning workloads and comes prepackaged with the best of Microsoft technologies for accelerated training, for example, OnnxRuntime Training (ORT), DeepSpeed, MSCCL, etc.
39
+
**Description**: The Azure Curated Environment for PyTorch is our latest PyTorch curated environment. It's optimized for large, distributed deep learning workloads and comes prepackaged with the best of Microsoft technologies for accelerated training (e.g., Onnx Runtime Training (ORT), DeepSpeed, MSCCL, etc.).
43
40
44
41
The following configurations are supported:
45
42
46
43
| Environment Name | OS | GPU Version| Python Version | PyTorch Version | ORT-training Version | DeepSpeed Version | torch-ort Version | Nebula Version |
Other packages like fairscale, horovod, msccl, protobuf, pyspark, pytest, pytorch-lightning, tensorboard, NebulaML, torchvision, torchmetrics to support all training needs
49
+
Other packages like fairscale, horovod, msccl, protobuf, pyspark, pytest, pytorch-lightning, tensorboard, NebulaML, torchvision, and torchmetrics are provided to support all training needs.
52
50
53
51
To learn more, see [Create custom ACPT curated environments](how-to-azure-container-for-pytorch-environment.md).
0 commit comments