Skip to content

Commit

Permalink
Add updated list of images
Browse files Browse the repository at this point in the history
  • Loading branch information
@sophies927
sophies927 committed May 29, 2024
1 parent 7de8a45 commit 742aad5
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 9 deletions.
24 changes: 15 additions & 9 deletions docs/ecosystem/acpt.md
View file
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
---
title: Azure Container for PyTorch (ACPT)
description: Learn more about Azure Container for PyTorch (ACPT) and how it utilizes ONNX Runtime
nav_order: 9
nav_order: 1
redirect_from: /docs/tutorials/ecosystem/acpt
---
# Azure Container for PyTorch (ACPT)
Expand All @@ -27,14 +27,20 @@ Azure Container for PyTorch (ACPT) is a lightweight, standalone environment that
## Supported configurations for Azure Container for PyTorch (ACPT)
The following configurations are supported in the Microsoft Container Registry (MCR):

Most recent image releases (last updated 4/25/2024):

| Environment Name | OS | GPU Version | Python Version | PyTorch Version | ORT-training Version | DeepSpeed Version | torch-ort Version | Nebula Version |
|-----------------------------|----------------|-------------|----------------|-----------------|----------------------|-------------------|-------------------|----------------|
| acpt-pytorch-2.2-cuda12.1 | Ubuntu 20.04 | cu121 | 3.10 | 2.2.2 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 |
| acpt-pytorch-2.1-cuda12.1 | Ubuntu 20.04 | cu121 | 3.10 | 2.1.2 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 |
| acpt-pytorch-2.0-cuda11.7 | Ubuntu 20.04 | cu117 | 3.10 | 2.0.1 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 |
| acpt-pytorch-1.13-cuda11.7 | Ubuntu 20.04 | cu117 | 3.10 | 1.13.1 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 |
| OS | GPU Type | Python Version | PyTorch Version | ORT-training version | DeepSpeed version | torch-ort Version | Nebula Version |
| - | - | - | - | - | - | - | - |
|ubuntu2004|cu117|3.8|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu117|3.9|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu117|3.10|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.8|2.0.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.10|2.0.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.8|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.10|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.8|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.10|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.10|2.3.0|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.10|2.3.0|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.8|2.1.2|1.18.0|0.14.2|1.17.0|0.16.13|

Other packages like fairscale, horovod, msccl, protobuf, pyspark, pytest, pytorch-lightning, tensorboard, NebulaML, torchvision, and torchmetrics are provided to support all training needs.

Expand Down
14 changes: 14 additions & 0 deletions docs/ecosystem/ptca_image_list.md
View file
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 | Image Name | MCR Image Name |
| - | - | - | - | - | - | - | - | - | - |
|ubuntu2004|cu117|3.8|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu117-py38-torch1131|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu117-py39-torch1131|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu117-py310-torch1131|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py38-torch201|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py310-torch201|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py38-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py310-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py38-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py310-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py310-torch230|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py310-torch230|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py38-torch212|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py38-torch212|

0 comments on commit 742aad5

Please sign in to comment.