-
Notifications
You must be signed in to change notification settings - Fork 3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
### Description Add an ACPT landing page to the ORT docs to publish supported configurations + most recent releases ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
- Loading branch information
1 parent
73df4e5
commit 65c3801
Showing
3 changed files
with
51 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,33 @@ | ||
--- | ||
title: Azure Container for PyTorch (ACPT) | ||
description: Learn more about Azure Container for PyTorch (ACPT) and how it utilizes ONNX Runtime | ||
nav_order: 1 | ||
redirect_from: /docs/tutorials/ecosystem/acpt | ||
--- | ||
# Azure Container for PyTorch (ACPT) | ||
{: .no_toc } | ||
|
||
Azure Container for PyTorch (ACPT) is a lightweight, standalone environment that includes needed components to effectively run optimized training for large models. It helps with reducing preparation costs and faster deployment time. ACPT can be used to quickly get started with various deep learning tasks with PyTorch on Azure. | ||
|
||
## Contents | ||
{: .no_toc } | ||
|
||
* TOC placeholder | ||
{:toc} | ||
|
||
|
||
## Why should I use ACPT? | ||
* **Flexibility:** Use as-is with preinstalled packages or build on top of the curated environment. | ||
* **Ease of use:** All components are installed and validated against dozens of Microsoft workloads to reduce setup costs and accelerate time to value. | ||
* **Efficiency:** Avoid unnecessary image builds and only have required dependencies that are accessible right in the image/container. | ||
* **Optimized training framework:** Set up, develop, and accelerate PyTorch models on large workloads, and improve training and deployment success rate. | ||
* **Up-to-date stack:** Access the latest compatible versions of Ubuntu, Python, PyTorch, CUDA/RocM, etc. | ||
* **Latest training optimization technologies:** Make use of ONNX Runtime, DeepSpeed, MSCCL, and more. | ||
|
||
## Supported configurations for Azure Container for PyTorch (ACPT) | ||
The following configurations are supported in the Microsoft Container Registry (MCR): [ptca_image_list.md](./ptca_image_list.md). | ||
|
||
Other packages like fairscale, horovod, msccl, protobuf, pyspark, pytest, pytorch-lightning, tensorboard, NebulaML, torchvision, and torchmetrics are provided to support all training needs. | ||
|
||
## Support | ||
Version updates for supported environments, including the base images they reference, are released every two weeks to address vulnerabilities no older than 30 days. Based on usage, some environments may be deprecated (hidden from the product but usable) to support more common machine learning scenarios. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
| OS | GPU Type | Python Version | PyTorch Version | ORT-training version | DeepSpeed version | torch-ort Version | Nebula Version | MCR Image Name | | ||
| - | - | - | - | - | - | - | - | - | | ||
|ubuntu2004|cu117|3.8|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py38-torch1131| | ||
|ubuntu2004|cu117|3.9|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py39-torch1131| | ||
|ubuntu2004|cu117|3.10|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py310-torch1131| | ||
|ubuntu2004|cu118|3.8|2.0.1|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py38-torch201| | ||
|ubuntu2004|cu118|3.10|2.0.1|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch201| | ||
|ubuntu2004|cu118|3.8|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py38-torch222| | ||
|ubuntu2004|cu118|3.10|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch222| | ||
|ubuntu2004|cu121|3.8|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py38-torch222| | ||
|ubuntu2004|cu121|3.10|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py310-torch222| | ||
|ubuntu2004|cu118|3.10|2.3.0|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch230| | ||
|ubuntu2004|cu121|3.10|2.3.0|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py310-torch230| | ||
|ubuntu2004|cu121|3.8|2.1.2|1.18.0|0.14.2|1.17.0|0.16.13|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py38-torch212| |