Skip to content

Commit afa1709

Browse files
authored
Merge pull request #268686 from sophies927/patch-1
Update resource-azure-container-for-pytorch.md with new configurations
2 parents 2412627 + 72899cb commit afa1709

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

articles/machine-learning/resource-azure-container-for-pytorch.md

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -22,33 +22,31 @@ Azure Container for PyTorch is a lightweight, standalone environment that includ
2222
2323
## Why should I use ACPT?
2424

25-
* Use as is with preinstalled packages or build on top of the curated environment.
26-
* Optimized training framework to set up, develop, accelerate PyTorch model on large workloads.
27-
* Up-to-date stack with the latest compatible versions of Ubuntu, Python, PyTorch, CUDA\RocM, etc.
28-
* Ease of use: All components installed and validated against dozens of Microsoft workloads to reduce setup costs and accelerate time to value.
29-
* Latest Training Optimization Technologies: [ONNX RunTime](https://onnxruntime.ai/) , [DeepSpeed](https://www.deepspeed.ai/), [MSCCL](https://github.com/microsoft/msccl),and others.
30-
* 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).
3633

3734
>[!IMPORTANT]
3835
> 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).
3936
4037
## Supported configurations for Azure Container for PyTorch (ACPT)
4138

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.).
4340

4441
The following configurations are supported:
4542

4643
| Environment Name | OS | GPU Version| Python Version | PyTorch Version | ORT-training Version | DeepSpeed Version | torch-ort Version | Nebula Version |
4744
| --- | --- | --- | --- | --- | --- | --- | --- | --- |
48-
|acpt-pytorch-2.0-cuda11.7|Ubuntu 20.04|cu117|3.8|2.0.1|1.15.1|0.9.5 |1.15.0|0.16.5|
49-
|acpt-pytorch-1.13-cuda11.7|Ubuntu 20.04|cu117|3.8|1.13.1|1.15.1|0.9.5|1.15.0|0.16.5|
45+
|acpt-pytorch-2.1-cuda12.1|Ubuntu 20.04|cu121|3.8|2.1.2|1.17.1|0.13.1|1.17.0|0.16.10|
46+
|acpt-pytorch-2.0-cuda11.7|Ubuntu 20.04|cu117|3.8|2.0.1|1.17.1|0.13.1|1.17.0|0.16.10|
47+
|acpt-pytorch-1.13-cuda11.7|Ubuntu 20.04|cu117|3.8|1.13.1|1.17.1|0.13.1|1.17.0|0.16.10|
5048

51-
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.
5250

5351
To learn more, see [Create custom ACPT curated environments](how-to-azure-container-for-pytorch-environment.md).
5452

0 commit comments

Comments
 (0)